mindspore_gs.pruner.PrunerFtCompressAlgo

.. py:class:: mindspore_gs.pruner.PrunerFtCompressAlgo(config=None)

    `PrunerFtCompressAlgo` 是CompAlgo的子类，实现了删除冗余卷积核并对网络进行完整训练的能力。

    参数：
        - **config** (dict) - 以字典的形式存放用于剪枝训练的配置，下面列出了受支持的属性：

          - **prune_rate** (float) - 值的取值范围是[0.0, 1.0)。

    异常：
        - **TypeError** - `prune_rate` 的数据类型不是 `float` 。
        - **ValueError** - `prune_rate` 小于0或者大于等于1。

    .. py:method:: apply(network, **kwargs)

        将Knockoff网络变为剪枝后的网络。

        参数：
            - **network** (Cell) - Knockoff网络。
            - **kwargs** (Dict) - 用于子类的可扩展入参。

        返回：
            返回剪枝后的网络。

        异常：
            - **TypeError** - `network` 不是Cell。

    .. py:method:: set_prune_rate(prune_rate)

        设置剪枝率。

        参数：
            - **prune_rate** (float) - 剪掉网络的大小。

        异常：
            - **TypeError** - `prune_rate` 不是float。
            - **ValueError** - `prune_rate` 小于0或者大于等于1。