ColossalAI/colossalai/zero/sharded_model
ver217 e6212f56cd
[hotfix] fix memory leak in backward of sharded model (#741)
2022-04-13 09:59:05 +08:00
..
__init__.py
_utils.py
reduce_scatter.py
sharded_model_v2.py [hotfix] fix memory leak in backward of sharded model (#741) 2022-04-13 09:59:05 +08:00
utils.py