Update the correctness verification and performance comparison tutorial

houseroad · houseroad · commit 126174779064 · 2017-11-23T09:35:55.000-08:00
diff --git a/tutorials/CorrectnessVerificationAndPerformanceComparison.ipynb b/tutorials/CorrectnessVerificationAndPerformanceComparison.ipynb
@@ -49,7 +49,7 @@
     "from caffe2.python import core\n",
     "from torch.autograd import Variable\n",
     "from onnx_caffe2.backend import Caffe2Backend\n",
-    "from onnx_caffe2.helper import c2_native_run_net, name_inputs, save_caffe2_net, load_caffe2_net, \\\n",
+    "from onnx_caffe2.helper import c2_native_run_net, save_caffe2_net, load_caffe2_net, \\\n",
     "    benchmark_caffe2_model, benchmark_pytorch_model\n",
     "\n",
     "\n",
@@ -231,7 +231,7 @@
     "pytorch_results = pytorch_model(*inputs)\n",
     "\n",
     "# Compute the results using the Caffe2 model.\n",
-    "_, caffe2_results = c2_native_run_net(init_net, predict_net, name_inputs(onnx_model, caffe2_inputs))"
+    "_, caffe2_results = c2_native_run_net(init_net, predict_net, caffe2_inputs)"
    ]
   },
   {
@@ -282,8 +282,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "PyTorch model's execution time is 0.637650489807 milliseconds/ iteration, 1568.25724434 iterations per second.\n",
-      "Caffe2 model's execution time is 0.43121188879 milliseconds / iteration, 2319.04552262 iterations per second\n"
+      "PyTorch model's execution time is 0.580716133118 milliseconds/ iteration, 1722.011742 iterations per second.\n",
+      "Caffe2 model's execution time is 0.399529695511 milliseconds / iteration, 2502.94286316 iterations per second\n"
      ]
     }
    ],
