InternLM/internlm/core/scheduler
chenxun.p 95488d8e8f update optimizer accumulate grad impl when fstp 2023-10-20 15:58:06 +08:00
..
__init__.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
base_scheduler.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
no_pipeline_scheduler.py update optimizer accumulate grad impl when fstp 2023-10-20 15:58:06 +08:00
pipeline_scheduler.py fix(pipeline): fix bugs for pipeline when enable mixed precision (#382) 2023-10-09 14:01:15 +08:00