InternLM/internlm/core
huangting4201 f3664bfbab
fix(train.py): fix scheduler metric hook skip error (#204)
2023-08-16 15:47:05 +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(train.py): fix scheduler metric hook skip error (#204) 2023-08-16 15:47:05 +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