设计
规格
API
API映射
迁移指南
FAQ
RELEASE NOTES
将MiniBatch切分成更细粒度的MicroBatch,用于流水线并行的训练中。
Note
micro_size必须大于或等于流水线stage的个数。
network (Cell) - 要修饰的目标网络。
micro_size (int) - MicroBatch大小。
Ascend GPU
Ascend
GPU
样例:
>>> net = Net() >>> net = PipelineCell(net, 4)