InternLM/internlm/core
Wenwen Qu e21ddc364b modify grad clipping due to moe 2023-08-09 15:04:19 +08:00
..
communication refactor(scheduler): rewrite pipeline scheduler (#138) 2023-08-03 11:48:12 +08:00
context modified: internlm/core/context/process_group_initializer.py 2023-08-08 15:59:12 +08:00
scheduler modify grad clipping due to moe 2023-08-09 15:04:19 +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