InternLM/internlm/core/scheduler
zhanglei 3a3ca71459 fix moe loss logger for the interleaved pp 2023-08-23 13:03:21 +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 fix moe loss logger for the interleaved pp 2023-08-23 13:03:21 +08:00