InternLM/internlm/core/scheduler
Wenwen Qu 916647c0a1
fix(pipeline): fix bugs for pipeline when enable mixed precision (#382)
* fix bugs for pipeline

* restore logic for empty fp32 group

* move optim.dtype to each param group
2023-10-09 14:01:15 +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 feat(moe): add moe module (#182) 2023-09-27 15:54:53 +08:00
pipeline_scheduler.py fix(pipeline): fix bugs for pipeline when enable mixed precision (#382) 2023-10-09 14:01:15 +08:00