InternLM/internlm/core
Qu Wenwen e1ecfa51ec Merge remote-tracking branch 'upstream/develop' into bugs/fix_mixed_precision_with_pipeline 2023-10-07 12:08:23 +08:00
..
communication Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
context feat(moe): add local data parallel support for experts (#376) 2023-09-28 13:38:02 +08:00
scheduler fix bugs for pipeline 2023-09-28 13:44:14 +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