mindspore.nn.PipelineCell

class mindspore.nn.PipelineCell(network, micro_size)

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.

Examples

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