解决方法参考网址:https://github.com/onnx/onnx/issues/1385
在使用命令torch.onnx.export时,加上keep_initializers_as_inputs=True,命令如下:
torch.onnx.export(model, img, "model.onnx", export_params=True, keep_initializers_as_inputs=True)
解决方法参考网址:https://github.com/onnx/onnx/issues/1385
在使用命令torch.onnx.export时,加上keep_initializers_as_inputs=True,命令如下:
torch.onnx.export(model, img, "model.onnx", export_params=True, keep_initializers_as_inputs=True)