From 01b2a4754fcb8192cf764c974c7360b6252f6164 Mon Sep 17 00:00:00 2001 From: Chin Huang Date: Sat, 18 Aug 2018 00:09:15 +0000 Subject: [PATCH] Fix runtime error for onnx_graph_to_caffe2_net Got 'The model does not have an ir_version set properly' exception when passing model.graph to onnx_graph_to_caffe2_net. The checker takes model parameter therefore pass model instead of graph. --- tutorials/PytorchCaffe2MobileSqueezeNet.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/PytorchCaffe2MobileSqueezeNet.ipynb b/tutorials/PytorchCaffe2MobileSqueezeNet.ipynb index b5ac4ba..d482677 100644 --- a/tutorials/PytorchCaffe2MobileSqueezeNet.ipynb +++ b/tutorials/PytorchCaffe2MobileSqueezeNet.ipynb @@ -258,7 +258,7 @@ "# Export to mobile\n", "from caffe2.python.onnx.backend import Caffe2Backend as c2\n", "\n", - "init_net, predict_net = c2.onnx_graph_to_caffe2_net(model.graph)\n", + "init_net, predict_net = c2.onnx_graph_to_caffe2_net(model)\n", "with open(\"squeeze_init_net.pb\", \"wb\") as f:\n", " f.write(init_net.SerializeToString())\n", "with open(\"squeeze_predict_net.pb\", \"wb\") as f:\n",