InternLM/internlm/train
yingtongxiong 62d193c763 add IS_SEQUENCE_PARALLEL check for norm module 2023-12-05 20:58:26 +08:00
..
__init__.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
training_internlm.py add IS_SEQUENCE_PARALLEL check for norm module 2023-12-05 20:58:26 +08:00
utils.py fix(moe): remove norm&gate force sync (#448) 2023-11-01 11:29:55 +08:00