update readme

Author: 545999961

examples/finetune/reranker/README.md

@@ -201,7 +201,7 @@ torchrun --nproc_per_node 2 \
     --output_dir ./test_decoder_only_base_bge-reranker-v2-minicpm-layerwise \
     --overwrite_output_dir \
     --learning_rate 2e-4 \
-    --fp16 \
+    --bf16 \
     --num_train_epochs 1 \
     --per_device_train_batch_size 2 \
     --gradient_accumulation_steps 1 \
@@ -260,7 +260,7 @@ torchrun --nproc_per_node 2 \
 	--output_dir ./test_decoder_only_base_bge-reranker-v2-minicpm-layerwise \
     --overwrite_output_dir \
     --learning_rate 2e-4 \
-    --fp16 \
+    --bf16 \
     --num_train_epochs 1 \
     --per_device_train_batch_size 2 \
     --gradient_accumulation_steps 1 \

examples/finetune/reranker/decoder_only/base.sh

@@ -46,7 +46,7 @@ training_args="\
     --output_dir ./test_decoder_only_base_bge-reranker-v2-gemma \
     --overwrite_output_dir \
     --learning_rate 2e-4 \
-    --fp16 \
+    --bf16 \
     --num_train_epochs $num_train_epochs \
     --per_device_train_batch_size $per_device_train_batch_size \
     --gradient_accumulation_steps $gradient_accumulation_steps \

examples/finetune/reranker/decoder_only/layerwise.sh

@@ -51,7 +51,7 @@ training_args="\
     --output_dir ./test_decoder_only_base_bge-reranker-v2-minicpm-layerwise \
     --overwrite_output_dir \
     --learning_rate 2e-4 \
-    --fp16 \
+    --bf16 \
     --num_train_epochs $num_train_epochs \
     --per_device_train_batch_size $per_device_train_batch_size \
     --gradient_accumulation_steps $gradient_accumulation_steps \