Former-commit-id: d86455f685fa531e651333e00b4fe54d895cf2e4
This commit is contained in:
hiyouga
2024-01-09 18:31:27 +08:00
parent 7c3311ad5a
commit 9537bc7f3e
9 changed files with 78 additions and 50 deletions

View File

@@ -457,7 +457,7 @@ deepspeed --num_gpus 8 --master_port=9901 src/train_bash.py \
"loss_scale_window": 1000,
"hysteresis": 2,
"min_loss_scale": 1
},
},
"zero_optimization": {
"stage": 2,
"allgather_partitions": true,