目前,我正在尝试修剪网络(简单的卷积自动编码器)-不用说没有成功。
首先,我指的是博客文章:
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。
这意味着我的过滤器已更新,尽管根据这篇文章,它不应该发生。
谁能告诉我我做错了什么(或如何正确做)?
最好
阿雷波