This commit is contained in:
richardodliu 2024-04-16 14:50:05 +08:00 committed by GitHub
commit 144924bf6d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -13,11 +13,11 @@
}, },
"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"
} }
}, },
@ -48,4 +48,4 @@
"train_batch_size": "auto", "train_batch_size": "auto",
"train_micro_batch_size_per_gpu": "auto", "train_micro_batch_size_per_gpu": "auto",
"wall_clock_breakdown": false "wall_clock_breakdown": false
} }