Design
Specification
API
API Mapping
Migration Guide
FAQ
RELEASE NOTES
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.
Ascend GPU
Ascend
GPU
Examples
>>> net = Net() >>> net = PipelineCell(net, 4)