github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
InternLM/internlm/solver/optimizer
History
chenxun.p 6682f5d92a fix reduce scatter async bug 2023-10-17 15:10:07 +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 fix reduce scatter async bug 2023-10-17 15:10:07 +08:00
store.py feat(moe):support zero for expert local dp (#404) 2023-10-09 17:45:26 +08:00
utils.py feat(moe):support zero for expert local dp (#404) 2023-10-09 17:45:26 +08:00