InternLM/internlm/solver/optimizer
zaglc 420e883d76 fix(fsdp): add mix-precision training 2023-09-07 17:11:01 +08:00
..
__init__.py feat(fsdp): add training option for fsdp 2023-09-04 18:01:30 +08:00
hybrid_zero_optim.py fix(fsdp): add mix-precision training 2023-09-07 17:11:01 +08:00
store.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
utils.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00