Update PytorchOnnxExport.ipynb (#18)

Shuolongbj · prasanthpul · commit c80f4f4ee56e · 2018-02-02T12:56:15.000-08:00
Update PytorchOnnxExport.ipynb
diff --git a/tutorials/PytorchOnnxExport.ipynb b/tutorials/PytorchOnnxExport.ipynb
@@ -109,7 +109,7 @@
     "# Obtain your model, it can be also constructed in your script explicitly\n",
     "model = torchvision.models.alexnet(pretrained=True)\n",
     "# Invoke export\n",
-    "torch.onnx.export(model, dummy_input, \"output/alexnet.onnx\")"
+    "torch.onnx.export(model, dummy_input, \"alexnet.onnx\")"
    ]
   },
   {
@@ -190,7 +190,7 @@
     "import onnx\n",
     "\n",
     "# Load the ONNX model\n",
-    "model = onnx.load(\"output/alexnet.onnx\")\n",
+    "model = onnx.load(\"alexnet.onnx\")\n",
     "\n",
     "# Check that the IR is well formed\n",
     "onnx.checker.check_model(model)\n",
