mirror of https://github.com/InternLM/InternLM
![]() * fix bugs for pipeline * restore logic for empty fp32 group * move optim.dtype to each param group |
||
---|---|---|
.. | ||
__init__.py | ||
base_scheduler.py | ||
no_pipeline_scheduler.py | ||
pipeline_scheduler.py |