mindspore/docs/api/api_python/nn/mindspore.nn.PipelineCell.rst

14 lines
413 B
ReStructuredText
Raw Normal View History

2022-01-25 14:28:23 +08:00
mindspore.nn.PipelineCell
=========================
2021-12-04 15:18:50 +08:00
.. py:class:: mindspore.nn.PipelineCell(network, micro_size)
将MiniBatch切分成更细粒度的MicroBatch用于流水线并行的训练中。
.. note::
micro_size必须大于或等于流水线stage的个数。
参数:
- **network** (Cell) - 要修饰的目标网络。
- **micro_size** (int) - MicroBatch大小。