InternLM/internlm/solver/optimizer
JiaoPL 646f1b45fa rm debug log 2023-10-13 12:25:46 +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 fix(optimizer/fsdp_optimizer.py): fsdp process empty params group (#408) 2023-10-10 20:06:04 +08:00
hybrid_zero_optim.py update layer norm to tensorboard 2023-10-13 12:07:58 +08:00
store.py feat(moe):support zero for expert local dp (#404) 2023-10-09 17:45:26 +08:00
utils.py rm debug log 2023-10-13 12:25:46 +08:00