ColossalAI/colossalai/nn/lr_scheduler
HELSON dceae85195
Added MoE parallel (#127)
2022-01-07 15:08:36 +08:00
..
__init__.py Support TP-compatible Torch AMP and Update trainer API (#27) 2021-11-18 19:45:06 +08:00
cosine.py Added MoE parallel (#127) 2022-01-07 15:08:36 +08:00
delayed.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
linear.py Support TP-compatible Torch AMP and Update trainer API (#27) 2021-11-18 19:45:06 +08:00
multistep.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
onecycle.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
poly.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
torch.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00