mindspore.nn.PipelineCell
- class mindspore.nn.PipelineCell(network, micro_size)[source]
Slice MiniBatch into finer-grained MicroBatch for use in pipeline-parallel training.
Note
micro_size must be greater or equal to pipeline stages.
- Supported Platforms:
Ascend
GPU
Examples
>>> import mindspore.nn as nn >>> # Define the network structure of LeNet5. Refer to >>> # https://gitee.com/mindspore/docs/blob/master/docs/mindspore/code/lenet.py >>> net = LeNet5() >>> net = nn.PipelineCell(net, 4)