huangting4201 4832671abe fix(pipeline_scheduler.py): fix tensor shape err and comm block (#210)		2023-08-21 12:09:27 +08:00
..
communication	fix(ci): fix ci train error (#199)	2023-08-15 20:09:54 +08:00
context	feat(*): support sequence_parallel (#180)	2023-08-07 16:42:52 +08:00
scheduler	fix(pipeline_scheduler.py): fix tensor shape err and comm block (#210)	2023-08-21 12:09:27 +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