InternLM/internlm/solver/optimizer
Wenwen Qu e21ddc364b modify grad clipping due to moe 2023-08-09 15:04:19 +08:00
..
__init__.py initial commit 2023-07-06 12:55:23 +08:00
hybrid_zero_optim.py modify grad clipping due to moe 2023-08-09 15:04:19 +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