mirror of https://github.com/hpcaitech/ColossalAI
6e51d296f0
* add zero1&2 optimizer * rename test ditectory * rename test files * change tolerance in test |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| base_store.py | ||
| bucket_store.py | ||
| gradient_store.py | ||
| parameter_store.py | ||
| tensor_bucket.py | ||