mindspore/docs/api/api_python/transformer/mindspore.nn.OpParallelConf...

9 lines
322 B
ReStructuredText
Raw Normal View History

.. py:class:: mindspore.nn.transformer.OpParallelConfig(data_parallel=1, model_parallel=1)
用于设置数据并行和模型并行的OpParallelConfig。
**参数:**
- **data_parallel** (int) - 表示数据并行度。默认值1
- **model_parallel** (int) - 表示模型并行度。默认值1