Update README_zh.md

2023-11-17 00:18:07 -08:00 · 2023-11-17 00:18:07 -08:00 · 7cab47b822
parent c9b499fa7e
commit 7cab47b822
1 changed files with 2 additions and 0 deletions
--- a/README_zh.md
+++ b/README_zh.md
@ -288,6 +288,8 @@ CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
    --per_device_train_batch_size 2 \
    --gradient_accumulation_steps 4 \
    --lr_scheduler_type cosine \
+    --top_k 0 \
+    --top_p 0.9 \
    --logging_steps 10 \
    --save_steps 1000 \
    --learning_rate 1e-5 \