15 Star 59 Fork 81

MindSpore/golden-stick

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindspore_gs.pruner.PrunerKfCompressAlgo.rst 1.14 KB
一键复制 编辑 原始数据 按行查看 历史
宦晓玲 提交于 2025-10-11 14:29 +08:00 . modify contents

mindspore_gs.pruner.PrunerKfCompressAlgo

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

    `PrunerKfCompressAlgo` 是CompAlgo的子类,实现了SCOP算法中利用高仿数据来学习发现冗余卷积核。

    .. note::
        - 针对入参 `config` ,目前 `PrunerKfCompressAlgo` 是没有可选的配置项,但为了兼容性, `config` 被保留,在初始化时以空字典代替。如 `kf_pruning = PrunerKfCompressAlgo({})` 。

    参数:
        - **config** (dict) - 算法配置参数。

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

        将网络变成Knockoff网络。

        参数:
            - **network** (Cell) - 原始待剪枝网络。
            - **kwargs** (Dict) - 用于子类的可扩展入参。

        返回:
            返回变换后的Knockoff网络。

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

    .. py:method:: callbacks(*args, **kwargs)

        定义SCOP剪枝算法特有的callbacks即生成高仿数据的callback。

        返回:
            SCOP剪枝算法的callbacks列表。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/golden-stick.git
git@gitee.com:mindspore/golden-stick.git
mindspore
golden-stick
golden-stick
master

搜索帮助