InternLM/internlm/solver/optimizer
huangting4201 9cf1ff0f6e feat(solver/optimizer/hybrid_zero_optim.py): minor update 2023-10-23 15:31:41 +08:00
..
__init__.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
base_optimizer.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
fsdp_optimizer.py feat(train): add fsdp training option (#293) 2023-10-09 18:59:31 +08:00
hybrid_zero_optim.py feat(solver/optimizer/hybrid_zero_optim.py): minor update 2023-10-23 15:31:41 +08:00
store.py feat(optimizer/hybrid_zero_optim.py): fix lint error 2023-10-20 16:22:29 +08:00
utils.py feat(moe):support zero for expert local dp (#404) 2023-10-09 17:45:26 +08:00