mindspore.parallel.nn.Pipeline
============================================================================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/parallel/mindspore.parallel.nn.Pipeline.rst
    :alt: 查看源文件


.. py:class:: mindspore.parallel.nn.Pipeline(network, micro_size, stage_config=None)

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

    参数：
        - **network** (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` 仍有剩余的配置项。