diff options
| author | Zitian Gao <zitian.gao@outlook.com> | 2025-05-29 19:20:42 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-05-29 19:20:42 +0800 |
| commit | 1fa309de8e63c6ac41d89505e451a9aff18ec49f (patch) | |
| tree | ae53717ee2aad0d91d87a9998224ec29708f7701 | |
| parent | 496a954c3aae0357f24d76f88383b2c36bd49277 (diff) | |
update params
| -rw-r--r-- | README.md | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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 \ |