mindspore.parallel.nn.Pipeline

class mindspore.parallel.nn.Pipeline(net, micro_size, stage_config=None)

指定流水线并行（pp）的micro_batch个数以及网络中各cell放到哪个stage去执行。

参数：

• net (Cell) - 将进行pp并行的网络。

• micro_size (int) - MicroBatchsize。

• stage_config (dict，可选) - 流水线并行对于每个cell的stage配置。默认值： None。

异常：

• TypeError - net 不是cell类型输入。

• TypeError - micro_size 不是整数类型。

• ValueError - micro_size 值异常，为0或者负数。

• KeyError - dict cell名称匹配异常，遍历当前 net 下所有 cell 仍有剩余的配置项。

支持平台：

Ascend

样例：

>>> from mindspore.parallel.nn import Pipeline
>>> # Define the network structure of LeNet5. Refer to
>>> # https://gitee.com/mindspore/docs/blob/master/docs/mindspore/code/lenet.py
>>> net = LeNet5()
>>> net = Pipeline(net, 4, stage_config={"cell_name_0": 0, "cell_name_1": 1})