More fix

dmlc · May 4, 2017 · 3b78554 · 3b78554
1 parent 8c15b09
commit 3b78554
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/how_to/predict.ipynb b/python/how_to/predict.ipynb
@@ -3405,7 +3405,7 @@
    },
    "outputs": [],
    "source": [
-    "mod = mx.mod.Module(symbol=sym, context=mx.gpu())"
+    "mod = mx.mod.Module(symbol=sym, label_names=None, context=mx.gpu())"
    ]
   },
   {
@@ -3609,7 +3609,7 @@
    "outputs": [],
    "source": [
     "batch_size = 32\n",
-    "mod2 = mx.mod.Module(symbol=sym, context=mx.gpu())\n",
+    "mod2 = mx.mod.Module(symbol=sym, label_names=None, context=mx.gpu())\n",
     "mod2.bind(for_training=False, data_shapes=[('data', (batch_size,3,224,224))])\n",
     "mod2.set_params(arg_params, aux_params, allow_missing=True)"
    ]
@@ -3725,7 +3725,7 @@
    "source": [
     "all_layers = sym.get_internals()\n",
     "sym3 = all_layers['flatten0_output']\n",
-    "mod3 = mx.mod.Module(symbol=sym3, context=mx.gpu())\n",
+    "mod3 = mx.mod.Module(symbol=sym3, label_names=None, context=mx.gpu())\n",
     "mod3.bind(for_training=False, data_shapes=[('data', (1,3,224,224))])\n",
     "mod3.set_params(arg_params, aux_params)\n"
    ]