Keras-过滤器修剪

时间:2018-07-30 18:09:51

标签: keras pruning

目前,我正在尝试修剪网络(简单的卷积自动编码器)-不用说没有成功。

首先,我指的是博客文章:

http://machinethink.net/blog/compressing-deep-neural-nets/

第一步,我只想将卷积滤波器设置为零,以使其不再起作用。到目前为止,我尝试过的是:

weights = model.get_weights() # weights is a numpy array
weights[2][0] = 0 # e.g. set the first filter of the second Conv2D-layer 0
weights[3][0] = 0 # and the filter's bias, of course

然后,我用这些权重初始化一个新模型:

newmodel.set_weights(weights)

并致电:

newmodel.fit()

但是,在获取权重之后:

newweights = newmodel.get_weights()

newweights[2][0] 

不再是0。

这意味着我的过滤器已更新,尽管根据这篇文章,它不应该发生。

谁能告诉我我做错了什么(或如何正确做)?

最好

阿雷波

0 个答案:

没有答案