Skip to content

Commit

Permalink
2023/07/19 add further pretraining code
Browse files Browse the repository at this point in the history
  • Loading branch information
michael-wzhu committed Jul 19, 2023
1 parent 32ecd0a commit 2eaac01
Showing 1 changed file with 41 additions and 0 deletions.
41 changes: 41 additions & 0 deletions src/further_ft/deepspeed_config_zero2.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
{
"fp16": {
"enabled": "auto",
"loss_scale": 0,
"loss_scale_window": 1000,
"initial_scale_power": 10,
"hysteresis": 2,
"min_loss_scale": 1
},
"optimizer": {
"type": "AdamW",
"params": {
"lr": "auto",
"weight_decay": "auto"
}
},
"scheduler": {
"type": "WarmupDecayLR",
"params": {
"warmup_min_lr": "auto",
"warmup_max_lr": "auto",
"warmup_num_steps": "auto",
"total_num_steps": "auto"
}
},
"zero_optimization": {
"stage": 2,
"allgather_partitions": true,
"allgather_bucket_size": 2e8,
"reduce_scatter": true,
"reduce_bucket_size": 2e8,
"overlap_comm": true,
"contiguous_gradients": true
},
"gradient_accumulation_steps": "auto",
"gradient_clipping": "auto",
"steps_per_print": 10,
"train_batch_size": "auto",
"train_micro_batch_size_per_gpu": "auto",
"wall_clock_breakdown": false
}

0 comments on commit 2eaac01

Please sign in to comment.