ColossalAI/colossalai/zero/sharded_model
ver217 0ef8819c67
polish docstring of zero (#612)
2022-04-01 14:50:56 +08:00
..
__init__.py
_utils.py [zero] add stateful tensor (#549) 2022-03-30 13:51:37 +08:00
reduce_scatter.py
sharded_model_v2.py polish docstring of zero (#612) 2022-04-01 14:50:56 +08:00
utils.py [polish] rename col_attr -> colo_attr (#558) 2022-03-31 12:25:45 +08:00