问题介绍:将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