InternLM/internlm/solver/optimizer
huangting4201 2e94870967 fix(internlm/train/training_internlm.py): remove set IS_TENSOR_PARALLEL attr 2023-10-08 20:22:40 +08:00
..
__init__.py feat(solver/optimizer): add new file fsdp_optimizer.py 2023-09-26 15:46:47 +08:00
base_optimizer.py feat(solver/optimizer): add new file fsdp_optimizer.py 2023-09-26 15:46:47 +08:00
fsdp_optimizer.py fix(internlm/train/training_internlm.py): remove set IS_TENSOR_PARALLEL attr 2023-10-08 20:22:40 +08:00
hybrid_zero_optim.py fix conflicts 2023-10-08 15:49:47 +08:00
store.py feat(moe): add local data parallel support for experts (#376) 2023-09-28 13:38:02 +08:00
utils.py feat(moe): add moe module (#182) 2023-09-27 15:54:53 +08:00