huangting4201 6b843253eb fix(optimizer/hybrid_zero_optim.py): remove redundant _accum_grad_buckets		2023-10-31 20:26:36 +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	fix(optimizer/hybrid_zero_optim.py): remove redundant _accum_grad_buckets	2023-10-31 20:26:36 +08:00
store.py	feat(optimizer/hybrid_zero_optim.py): fix lint error	2023-10-20 16:22:29 +08:00
utils.py	feat(optimizer): add layer norm to tensorboard (#429)	2023-10-23 17:07:04 +08:00