hiyouga
/

Baichuan-7B-sft

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

hiyouga commited on Sep 10, 2023

Commit

7d68667

•

1 Parent(s): 88eba19

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -75,7 +75,7 @@ CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
     --learning_rate 5e-5 \
     --max_grad_norm 0.5 \
     --num_train_epochs 2.0 \
-    --dev_ratio 0.01 \
     --evaluation_strategy steps \
     --load_best_model_at_end \
     --plot_loss \

     --learning_rate 5e-5 \
     --max_grad_norm 0.5 \
     --num_train_epochs 2.0 \
+    --val_size 0.01 \
     --evaluation_strategy steps \
     --load_best_model_at_end \
     --plot_loss \