角膜外科医生
一个用于在经过训练的Keras模型上进行网络手术的库。 对于深度神经网络修剪很有用。
Keras外科医生提供了简单的方法来修改训练有素的模型。 当前实现了以下功能:
从图层中删除神经元/通道
删除图层
插入图层
替换图层
Keras-Surgeon与任何模型架构兼容。 一次遍历网络可以修改任意数量的层。
这些修改有时被称为网络手术,这激发了该软件包的名称。
背景
这个项目的动机是我对深度学习的兴趣以及对尝试使用研究文献中已经了解的一些修剪方法进行试验的愿望。
我创建此程序包是因为找不到从Keras模型中修剪神经元的简单方法。 希望对其他人有用。
安装
使用pip从安装Keras-Surgeon。
pip install kerassurgeon
如果您要安装示例的依赖项,请执行以下操作:
pip install kerassurgeon[examples]
它
1