InternLM/internlm/utils
cx 0268d8eda1
refactor(scheduler): rewrite pipeline scheduler (#138)
* refactor(scheduler): rewrite pipeline scheduler

* fix(*): fix pipeline scheduler bugs

* fix(*): fix merge bug

* feat(*): update codes with todo tag

* feat(*): add comments

* feat(internlm/core/scheduler): update recv_prev/next logic

* feat(utils/evaluation.py): update sche metric hook for valid

---------

Co-authored-by: huangting.p <huangting@sensetime.com>
2023-08-03 11:48:12 +08:00
..
__init__.py initial commit 2023-07-06 12:55:23 +08:00
checkpoint.py initial commit 2023-07-06 12:55:23 +08:00
common.py feat(core/scheduler): support pipeline parallel (#98) 2023-07-24 20:52:09 +08:00
evaluation.py refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
logger.py feat(utils/evaluation.py): support evaluate (#154) 2023-08-02 19:03:59 +08:00
megatron_timers.py initial commit 2023-07-06 12:55:23 +08:00
model_checkpoint.py feat(core/scheduler): support pipeline parallel (#98) 2023-07-24 20:52:09 +08:00
parallel.py feat(utils/logger.py): support uniscale logger (#152) 2023-08-01 17:37:32 +08:00
registry.py feat(core/scheduler): support pipeline parallel (#98) 2023-07-24 20:52:09 +08:00
storage_manager.py initial commit 2023-07-06 12:55:23 +08:00
writer.py feat(utils/logger.py): support uniscale logger (#152) 2023-08-01 17:37:32 +08:00