github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-28 13:05:26 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
88759e289efd0a7b5e0d7bf8e01dbe29db85cf71
ColossalAI/colossalai/zero/comm/__init__.py
HELSON 88759e289e [zero] add ZeroTensorShardStrategy (#793)
2022-04-19 14:32:45 +08:00

2 lines
33 B
Python
Raw Blame History

from .zero_comm import ZeroDist
Reference in New Issue View Git Blame Copy Permalink