InternLM/internlm/core
Wenwen Qu 629e6a5ad1 add comments for moe 2023-08-25 19:03:31 +08:00
..
communication fix(ci): fix ci train error (#199) 2023-08-15 20:09:54 +08:00
context add comments for moe 2023-08-25 19:03:31 +08:00
scheduler add comments for moe 2023-08-25 19:03:31 +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 Merge branch 'develop' into feature_add_moe 2023-08-25 13:35:56 +08:00