Wenwen Qu 3c992a2101 fix(pipeline): fix interleave type assert and metrics error (#423) ... * fix interleave type assert bug * refactor code for assert * fix is_no_pp_or_last_stage logic		2023-10-19 17:29:30 +08:00
..
communication	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
context	fix(pipeline): fix interleave type assert and metrics error (#423)	2023-10-19 17:29:30 +08:00
scheduler	fix(pipeline): fix interleave type assert and metrics error (#423)	2023-10-19 17:29:30 +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	feat(moe): add moe module (#182)	2023-09-27 15:54:53 +08:00
trainer.py	feat(moe): add moe module (#182)	2023-09-27 15:54:53 +08:00