mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
12 lines
171 B
12 lines
171 B
3 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- encoding: utf-8 -*-
|
||
|
|
|
||
|
|
parallel = dict(
|
||
|
|
pipeline=dict(size=2),
|
||
|
|
tensor=dict(
|
||
|
|
size=8,
|
||
|
|
depth=2,
|
||
|
|
mode='2.5d'
|
||
|
|
)
|
||
|
|
)
|