InternLM/internlm/solver/optimizer
Wenwen Qu 929dd36cf2 avoid moe parameter partition in zero optimizer 2023-08-15 17:52:51 +08:00
..
__init__.py initial commit 2023-07-06 12:55:23 +08:00
hybrid_zero_optim.py avoid moe parameter partition in zero optimizer 2023-08-15 17:52:51 +08:00
store.py initial commit 2023-07-06 12:55:23 +08:00
utils.py refactor(*): refactor the code with no-apex (#170) 2023-08-03 11:24:12 +08:00