This commit is contained in:
hiyouga
2024-01-09 18:31:27 +08:00
parent ebee4f6a2a
commit 4571068e1e
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,