如何重复使用火炬到Keras的砝码

2024-05-13 19:35:16 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个下载的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


Tags: 模型gt网络numpy架构空间数组keras