mirror of https://github.com/InternLM/InternLM
* fix interleave type assert bug * refactor code for assert * fix is_no_pp_or_last_stage logic |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| base_scheduler.py | ||
| no_pipeline_scheduler.py | ||
| pipeline_scheduler.py | ||