From 7cab47b822ddf7f94d2ccfbe2ba4836e35f104f4 Mon Sep 17 00:00:00 2001 From: Yuchen Han <42163912+hannlp@users.noreply.github.com> Date: Fri, 17 Nov 2023 00:18:07 -0800 Subject: [PATCH] Update README_zh.md --- README_zh.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README_zh.md b/README_zh.md index dbcee80a..f03f27f0 100644 --- 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 \