PaddleHub 下载的预训练模型,使用 `paddle.load()` 无法加载?

9avjhtql  于 6个月前  发布在  其他
关注(0)|答案(4)|浏览(49)
  1. 使用paddlehub下载了vgg19_imagenet的预训练模型
  2. 使用paddle.load()加载权重
  3. 报错如下

  1. 我在aiStudio上的在线环境中查看预训练权重,没有文件类型吗?

li9yvcax

li9yvcax1#

  1. 使用paddlehub下载了vgg19_imagenet的预训练模型
  2. 使用paddle.load()加载权重
  3. 报错如下

  1. 我在aiStudio上的在线环境中查看预训练权重,没有文件类型吗?

你圈起来的文件的大小为0,检查一下是否正常下载了。

disho6za

disho6za2#

你想通过hub下载模型文件,然后使用paddle.load()加载模型吗?如果是要直接用于finetune或者预测的话,建议使用hub.Module的方式加载。

goucqfw6

goucqfw63#

我确实想通过hub下载模型文件,然后使用paddle.load()加载模型。但是我需要获得模型多尺度的feature map输出,而不仅仅是获取模型最后的输出。

rfbsl7qr

rfbsl7qr4#

我确实是想通过hub下载模型文件,自己使用paddle.load()加载模型,但是我需要获得模型多尺度的feature map输出, 不是只获取模型最后的输出。
如果module目前没提供获取feature map的输出,你需要仿照module里的方式,加载预训练模型参数,然后在前向传播中获取你想要的中间层输出。
resnet152_vd_imagenet 举例子,你可以在forward里获取到feature后返回
PaddleHub/modules/image/classification/resnet152_vd_imagenet/module.py
Line 252 in 778db1a
| | y=paddle.reshape(y, shape=[-1, self.pool2d_avg_channels]) |

相关问题