Toybrick

senet154 toolkit1.3.0 segmentation fault

jy_dll

注册会员

积分
122
发表于 2020-2-27 18:40:42    查看: 7467|回复: 3 | [复制链接]    打印 | 显示全部楼层
本帖最后由 jy_dll 于 2020-2-27 18:52 编辑

基于rknn-toolkit-1.3.0 尝试转换pytorch senet154, 用load_pytorch 导入*.pt, 导入成功,但是build失败,log如附件所示,senet154如附件所示,麻烦帮忙看看,非常感谢~
env:torch1.2.0, torchvision0.4.0, tensorflow1.14
不开量化,不预编译
------------------------------------------------------------------------
D Fill zero bias for convolution_at_input14.1_810 (convolution)
D Fill zero bias for convolution_at_input7.1_812 (convolution)
D Fill zero bias for convolution_at_input26.1_813 (convolution)
D Fill zero bias for convolution_at_input11.1_818 (convolution)
D Fill zero bias for convolution_at_input4.1_819 (convolution)
D Fill zero bias for convolution_at_input1.1_822 (convolution)
D Optimizing network with conv2d_big_kernel_size_transform
done
--> Building model..
I Generate input meta
I Load input meta
I Generate input meta
D import clients finished
I Load net...
Segmentation fault (core dumped)





本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

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

使用道具 举报

jefferyzhang

版主

积分
12919
发表于 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
回复

使用道具 举报

wangkelin

注册会员

积分
114
发表于 2021-9-3 10:57:27 | 显示全部楼层
se模块不是不支持吗?自定义算子不是只支持tf吗,为什么还能加载.pt文件
回复

使用道具 举报

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

本版积分规则

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


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