github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/shardformer/test/config.py

2 lines
68 B
Python
Raw Blame History

parallel = dict(data=1, pipeline=1, tensor=dict(size=2, mode='1d'))
Reference in New Issue View Git Blame Copy Permalink