由pytorch到onnx转换efficientnet,onnx引用转换好的模型时报错:
line 55, in optimize
optimized_model_str = C.optimize(model_str, passes)
IndexError: _Map_base::at
解决方法:
这是pytorch1.3.1新添加的接口:
https://pytorch.org/docs/stable/onnx.html?highlight=keep_initializers_as_inputs
在export时添加 keep_initializers_as_inputs 参数:
torch.onnx.export(model, dummy_input, “test-b0.onnx”, export_params=True, keep_initializers_as_inputs=True, verbose=True)
在pytorch1.2版本中,不需要使用这个initializer接口,直接export就可以。
引用:
https://github.com/onnx/onnx/issues/1385