Toybrick

senet154 toolkit1.3.0 segmentation fault

jefferyzhang

版主

积分
12956
楼主
发表于 2020-2-28 08:52:46 | 显示全部楼层
已上报问题,NPU部门有回复我会回给你
回复

使用道具 举报

jefferyzhang

版主

积分
12956
沙发
发表于 2020-2-28 17:40:42 | 显示全部楼层
NPU部门回复:

env: torch1.2.0, torchvision0.4.0, tensorflow1.14
我这边试了一下,并未出现段错误,float的pytorch的推理结果和rknn的推理结果是一样的。
请按照如下步骤排查:
1.生成pt是否正确,是否忘记调用net.eval()?
net = senet.senet154()
net.eval()
dummy_input = torch.randn([1, 3, 224, 224])
trace_model = torch.jit.trace(net, dummy_input)
trace_model.save(pt_save_path)
2. 固件或驱动是否是最新版本?
3. 使用rknn_toolkit v1.3.1 beta3:
链接: https://pan.baidu.com/s/1Kn2FGAdF_j3CMLNEsC3OPw 提取码: rcds
回复

使用道具 举报

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

本版积分规则

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


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