Toybrick

楼主: zhw

pt转rknn出错

zhw

注册会员

积分
103
 楼主| 发表于 2020-12-30 10:02:31 | 显示全部楼层
jefferyzhang 发表于 2020-12-30 09:55
为何需要['model'].eval()
直接eval的模型无法推理么?

eval不是转换成推理模式吗?我去掉.eval()也可以推理成功
回复

使用道具 举报

jefferyzhang

超级版主

积分
7316
发表于 2020-12-30 10:06:41 | 显示全部楼层
zhw 发表于 2020-12-30 10:02
eval不是转换成推理模式吗?我去掉.eval()也可以推理成功

我是说['model']
回复

使用道具 举报

zhw

注册会员

积分
103
 楼主| 发表于 2020-12-30 10:10:18 | 显示全部楼层

    model = torch.load(weights, map_location=device).eval()
AttributeError: 'dict' object has no attribute 'eval'
回复

使用道具 举报

jefferyzhang

超级版主

积分
7316
发表于 2020-12-30 10:15:50 | 显示全部楼层
zhw 发表于 2020-12-30 10:10
model = torch.load(weights, map_location=device).eval()
AttributeError: 'dict' object has no a ...

说明你的模型并不是通用模型。
rknn也是得先读取torch模型才能转换的,都不能用常规方法读出来,转换肯定会出问题。

pytorch我并不是很熟,还是建议你qq群里问下NPU部门的人,你这个问题解决后如果还不行
可以试着把模型文件@给他们让他们调试下
回复

使用道具 举报

zhw

注册会员

积分
103
 楼主| 发表于 2020-12-30 10:19:49 | 显示全部楼层
jefferyzhang 发表于 2020-12-30 10:15
说明你的模型并不是通用模型。
rknn也是得先读取torch模型才能转换的,都不能用常规方法读出来,转换肯定 ...

好的,万分感谢!!!
回复

使用道具 举报

zhw

注册会员

积分
103
 楼主| 发表于 2020-12-30 10:31:26 | 显示全部楼层
jefferyzhang 发表于 2020-12-30 10:15
说明你的模型并不是通用模型。
rknn也是得先读取torch模型才能转换的,都不能用常规方法读出来,转换肯定 ...

weights = 'plate.pt'
model = torch.load(weights, map_location=device)['model'].float().eval()
torch.save(model, 'plate1.pt')
我尝试着只保存['model'],还是不行
大佬,群里谁是npu部门的人?
回复

使用道具 举报

jefferyzhang

超级版主

积分
7316
发表于 2020-12-30 10:54:31 | 显示全部楼层
zhw 发表于 2020-12-30 10:31
weights = 'plate.pt'
model = torch.load(weights, map_location=device)['model'].float().eval()
torc ...

群管理员都是
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表