InternLM/internlm/core/scheduler
zhanglei 72e3b1afd5 change the scale position for latent moe_loss 2023-08-23 13:25:20 +08:00
..
__init__.py refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
base_scheduler.py fix(train.py): fix scheduler metric hook skip error (#204) 2023-08-16 15:47:05 +08:00
no_pipeline_scheduler.py add logger for moe_loss 2023-08-17 16:52:11 +08:00
pipeline_scheduler.py change the scale position for latent moe_loss 2023-08-23 13:25:20 +08:00