InternLM/internlm/train
zigzagcai f8eaf618af revert worker_init_fn since we move gc disable after dataloader launching 2023-12-29 19:34:38 +08:00
..
__init__.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
training_internlm.py revert worker_init_fn since we move gc disable after dataloader launching 2023-12-29 19:34:38 +08:00
utils.py fix(moe): remove norm&gate force sync (#448) 2023-11-01 11:29:55 +08:00