mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-07 06:32:06 +00:00
11 lines
152 B
Python
11 lines
152 B
Python
#!/usr/bin/env python
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
parallel = dict(
|
|
pipeline=dict(size=2),
|
|
tensor=dict(
|
|
size=8,
|
|
mode='3d'
|
|
)
|
|
)
|