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

5 lines
93 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