mindspore.nn.MicroBatchInterleaved

class mindspore.nn.MicroBatchInterleaved(network, interleave_num=2)[source]

Wrap the network with Batch Size.

Parameters

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

  • interleave_num (int) – split num of batch size. Default: 2.

Supported Platforms:

Ascend GPU

Examples

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