我有一个下载的Torch7模型(在Lua开发)已经过培训,我想在同一个网络架构中使用权重,但在Keras中
我可以使用Torchfile检查预先训练好的模型,这样我就知道了层的形状,并且可以访问numpy数组
但是权重在Lua和Keras中似乎有不同的维度,例如:
在Conv2D(Lua中的空间卷积)层之一中:
预先训练的重量-->;形状:(256、48、5、5)
期望从Keras中重新创建的模型-->;形状:(5,5,96,256)
256是筛选器数
96是输入的通道数
5,5是内核大小
48岁
所以我不能使用set_weights()
函数
要在此处运行的代码:Google Drive
目前没有回答
相关问题 更多 >
编程相关推荐