InternLM/internlm/core/scheduler
Wenwen Qu e21ddc364b modify grad clipping due to moe 2023-08-09 15:04:19 +08:00
..
__init__.py refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
base_scheduler.py refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
no_pipeline_scheduler.py modify grad clipping due to moe 2023-08-09 15:04:19 +08:00
pipeline_scheduler.py feat(*): support sequence_parallel (#180) 2023-08-07 16:42:52 +08:00