You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/nn/lr_scheduler
Baizhou Zhang 39f2582e98
[hotfix] fix lr scheduler bug in torch 2.0 (#4864)
1 year ago
..
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
cosine.py [misc] add last_epoch in CosineAnnealingWarmupLR (#4778) 1 year ago
delayed.py [hotfix] fix lr scheduler bug in torch 2.0 (#4864) 1 year ago
linear.py [misc] update pre-commit and run all files (#4752) 1 year ago
multistep.py [misc] update pre-commit and run all files (#4752) 1 year ago
onecycle.py [misc] update pre-commit and run all files (#4752) 1 year ago
poly.py [misc] update pre-commit and run all files (#4752) 1 year ago
torch.py