changed config

sotopia-lab · Nov 7, 2023 · 2441021 · 2441021
1 parent 9db6bbe
commit 2441021
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_rl/finetune-llama-2-13b.sh b/llm_rl/finetune-llama-2-13b.sh
@@ -27,7 +27,7 @@ deepspeed src/train_bash.py \
     --save_strategy "epoch" \
     --save_total_limit 5 \
     --learning_rate 5e-5 \
-    --num_train_epochs 10.0 \
+    --num_train_epochs 15.0 \
     --plot_loss \
     --bf16 \
     --deepspeed ./deepspeed_config_s2.json \