fix caffe importer (#5415)

apache · Mar 15, 2017 · 2f201fa · 2f201fa
1 parent ec19e4d
commit 2f201fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/caffe_converter/caffe_parser.py b/tools/caffe_converter/caffe_parser.py
@@ -42,7 +42,7 @@ def read_caffemodel(prototxt_fname, caffemodel_fname):
         return (layers, layer_names)
     else:
         proto = caffe_pb2.NetParameter()
-        with open(fname, 'rb') as f:
+        with open(caffemodel_fname, 'rb') as f:
             proto.ParseFromString(f.read())
         return (get_layers(proto), None)