首页 Paddle框架 帖子详情
torch预训练模型转换为PaddlePaddle模型时遇到问题
收藏
快速回复
Paddle框架 问答模型训练 93 4
torch预训练模型转换为PaddlePaddle模型时遇到问题
收藏
快速回复
Paddle框架 问答模型训练 93 4

预训练模型转换时遇到此问题,求解答。

torch parm fc.weight not exist in paddle modle
torch parm fc.bias not exist in paddle modle

0
收藏
回复
全部评论(4)
三岁
#2 回复于2021-10-12

我记得是需要经过第三方转换的,可以去查看一下,如果不行可以去官方qq里面咨询一下

0
回复
w
wheart1235
#3 回复于2021-10-13
三岁 #2
我记得是需要经过第三方转换的,可以去查看一下,如果不行可以去官方qq里面咨询一下

谢谢

0
回复
s
stoneLi
#4 回复于2021-10-17

安装好paddle环境

在paddle里面安装torch

然后,

import paddle
import torch
import numpy as np

name = 'vgg16net'
outpath = './paddlePretrained/'
model = torch.load('./vgg16net.pth')
print(model.keys())    #输出key
weight_dict = {}
weight_dict.setdefault('conv_block_1_conv1.weight', model['conv1_1.weight'].numpy())

....

weight_dict.setdefault('conv_block_5_conv3.bias', model['conv5_3.bias'].numpy())
print(weight_dict.keys())   # 查看weight_dict中的key
paddle.save(weight_dict, outpath + 'vgg16net.pdparams')

0
回复
s
stoneLi
#5 回复于2021-10-17
stoneLi #4
安装好paddle环境 在paddle里面安装torch 然后, import paddle import torch import numpy as np name = 'vgg16net' outpath = './paddlePretrained/' model = torch.load('./vgg16net.pth') print(model.keys())    #输出key weight_dict = {} weight_dict.setdefault('conv_block_1_conv1.weight', model['conv1_1.weight'].numpy()) .... weight_dict.setdefault('conv_block_5_conv3.bias', model['conv5_3.bias'].numpy()) print(weight_dict.keys())   # 查看weight_dict中的key paddle.save(weight_dict, outpath + 'vgg16net.pdparams')
展开

然后按paddle的导入,正常导入使用

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户