You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/zero/sharded_optim/bookkeeping
HELSON d565a24849
[zero] add unit testings for hybrid parallelism (#2486)
2 years ago
..
__init__.py [zero] migrate zero1&2 (#1878) 2 years ago
base_store.py [zero] polish low level optimizer (#2473) 2 years ago
bucket_store.py [zero] add unit testings for hybrid parallelism (#2486) 2 years ago
gradient_store.py [zero] migrate zero1&2 (#1878) 2 years ago
parameter_store.py [zero] polish low level optimizer (#2473) 2 years ago
tensor_bucket.py [zero] migrate zero1&2 (#1878) 2 years ago