torch预训练模型转换为PaddlePaddle模型时遇到问题
收藏
预训练模型转换时遇到此问题,求解答。
torch parm fc.weight not exist in paddle modle
torch parm fc.bias not exist in paddle modle
0
收藏
请登录后评论
我记得是需要经过第三方转换的,可以去查看一下,如果不行可以去官方qq里面咨询一下
谢谢
安装好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的导入,正常导入使用