refactor CosineWarmupLr, add test, add get_cosine_warmup_scheduler - Githubissues

PistonY / torch-toolbox

🛠 Toolbox to extend PyTorch functionalities

BSD 3-Clause "New" or "Revised" License

416 stars 56 forks source link

refactor CosineWarmupLr, add test, add get_cosine_warmup_scheduler #17

Closed garyhlai closed 2 years ago

garyhlai commented 2 years ago

refactor CosineWarmupLr for readability
add get_cosine_warmup_scheduler to support multi param groups
add test for CosineWarmupLr and get_cosine_warmup_scheduler

PistonY commented 2 years ago

LGTM😚