github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
InternLM/internlm/core
History
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
..
communication refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
context initial commit 2023-07-06 12:55:23 +08:00
scheduler refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00
engine.py initial commit 2023-07-06 12:55:23 +08:00
gradient_handler.py initial commit 2023-07-06 12:55:23 +08:00
naive_amp.py refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
trainer.py feat(core/scheduler): support pipeline parallel (#98) 2023-07-24 20:52:09 +08:00