如何从火炬到Keras重用权重

时间:2019-07-16 17:41:46

标签: python tensorflow keras lua torch

我有一个经过培训的下载的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

0 个答案:

没有答案