onnx调用报错IndexError: _Map_base::at

问题介绍:将pytorch模型转换为onnx,调用onnx报错IndexError: _Map_base::at

原因分析:pytorch>=1.3,模型转换为onnx并导出时,torch.onnx.export()函数发生变换,需要设置参数keep_initializers_as_inputs

torch.onnx.export(model, input, 'model.onnx', verbose=True, keep_initializers_as_inputs=True)

pytorch<1.3则不需要设置参数keep_initializers_as_inputs

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读