|
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 |
|