From c761a9987c3b040d07a9b1f6c31ecabc372b2e96 Mon Sep 17 00:00:00 2001 From: Yuchen Han <42163912+hannlp@users.noreply.github.com> Date: Fri, 17 Nov 2023 00:17:36 -0800 Subject: [PATCH] Update README.md Former-commit-id: c9b499fa7e1de6ff3467a3c6bf44e559deb492f7 --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 7ba2b748..88a820da 100644 --- a/README.md +++ b/README.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 \