MindSpore Python API
MindSpore C++ API
Wrap the network with Micro Batch.
Note
micro_size must be greater or equal to pipeline stages.
network (Cell) – The target network to wrap.
micro_size (int) – MicroBatch size.
Examples
>>> net = Net() >>> net = PipelineCell(net, 4)