Added explicit 'NHWC' order for all image-related operators. Also rem…

…oved unnecessary write-back in the notebook.
suraj-deshmukh · Oct 26, 2015 · fe2ee9a · fe2ee9a
1 parent ed00c01
commit fe2ee9a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/multibox.ipynb b/multibox.ipynb
@@ -84,8 +84,6 @@
     "# net is the network definition.\n",
     "net = caffe2_pb2.NetDef()\n",
     "net.ParseFromString(open('multibox_net.pb').read())\n",
-    "with open('multibox_net.pb', 'w') as fid:\n",
-    "    fid.write(net.SerializeToString())\n",
     "\n",
     "# tensors contain all the parameters used in the net.\n",
     "# The multibox model is relatively large so we have stored the parameters in multiple files.\n",

diff --git a/multibox_net.pb b/multibox_net.pb