fixes #186

microsoft · May 11, 2018 · ec613d9 · ec613d9
1 parent 6604534
commit ec613d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmdnn/conversion/caffe/saver.py b/mmdnn/conversion/caffe/saver.py
@@ -4,6 +4,8 @@
 def save_model(MainModel, network_filepath, weight_filepath, dump_filepath):
     dump_net = dump_filepath + '.prototxt'
     dump_weight = dump_filepath + '.caffemodel'
+    dump_net = str(dump_net)
+    dump_weight = str(dump_weight)
     MainModel.make_net(dump_net)
     MainModel.gen_weight(weight_filepath, dump_weight, dump_net)
     model = caffe.Net(dump_net, dump_weight, caffe.TEST)