mindspore.nn.PipelineCell

class mindspore.nn.PipelineCell(network, micro_size)[source]

Wrap the network with Micro Batch.

Note

micro_size must be greater or equal to pipeline stages.

Parameters

  • network (Cell) – The target network to wrap.

  • micro_size (int) – MicroBatch size.

Supported Platforms:

Ascend GPU

Examples

>>> net = Net()
>>> net = PipelineCell(net, 4)