modify the "scheduler" in "finetune/configs/ds_config_zero3.json" to be consistent with the sample shell script to finetune

This commit is contained in:
richardodliu 2024-03-15 12:14:16 +00:00
parent fd2db2cafe
commit 05e0ba01aa

View File

@ -13,11 +13,12 @@
}, },
"scheduler": { "scheduler": {
"type": "WarmupLR", "type": "WarmupCosineLR",
"params": { "params": {
"warmup_min_lr": "auto", "total_num_steps": "auto",
"warmup_max_lr": "auto", "warmup_num_steps": "auto",
"warmup_num_steps": "auto" "warmup_type": "linear",
"warmup_min_ratio": 0.1
} }
}, },