Skip to content

Commit bb53a16

Browse files
authored
add a basic ds zero3 config (axolotl-ai-cloud#347)
better defaults for ds
1 parent 10405b9 commit bb53a16

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

ds_config.json deepspeed/zero3.json

+5-5
Original file line numberDiff line numberDiff line change
@@ -37,18 +37,18 @@
3737
"lr": "auto",
3838
"betas": [
3939
0.9,
40-
0.999
40+
0.95
4141
],
4242
"eps": 1e-8,
4343
"weight_decay": "auto"
4444
}
4545
},
4646
"scheduler": {
47-
"type": "OneCycle",
47+
"type": "WarmupLR",
4848
"params": {
49-
"cycle_min_lr": 0.00001,
50-
"cycle_max_lr": 0.00003,
51-
"cycle_first_step_size": 120
49+
"warmup_min_lr": "auto",
50+
"warmup_max_lr": "auto",
51+
"warmup_num_steps": "auto"
5252
}
5353
},
5454
"train_batch_size": "auto",

0 commit comments

Comments
 (0)