From 1fa309de8e63c6ac41d89505e451a9aff18ec49f Mon Sep 17 00:00:00 2001 From: Zitian Gao Date: Thu, 29 May 2025 19:20:42 +0800 Subject: update params --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index a086988..5fde44e 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ accelerate launch train.py \ --micro_batch_size auto \ --temperature 0.5 \ --learning_rate 2e-5 \ - --max_steps 1000 \ + --max_steps 50 \ --log_steps 1 \ --save_steps 1 \ --run_name one_shot \ @@ -42,7 +42,7 @@ accelerate launch train.py \ --micro_batch_size auto \ --temperature 0.5 \ --learning_rate 2e-5 \ - --max_steps 1000 \ + --max_steps 50 \ --log_steps 1 \ --save_steps 1 \ --run_name multi_shot \ -- cgit v1.2.3