我有一个经过培训的下载的Torch7模型(由Lua开发),我想在Keras的相同网络体系结构中使用权重。
我可以使用Torchfile检查预先训练的模型,因此我知道了层的形状,并且可以访问numpy数组。
但是重量在Lua和Keras中似乎有不同的尺寸,例如:
在Conv2D(Lua中的SpatialConvolution)层之一中:
预训练砝码->形状:(256、48、5、5)
从Keras重建的模型中获得的期望->形状:(5、5、96、256)
过滤器数是256个
96是输入的通道数
5,5是内核大小
48吗?
所以我不能使用set_weights()
函数。
要在此处运行的代码:Google Drive