fixed main() of convert_symbol.py(#4121) (#4131)

apache · Dec 7, 2016 · fc874df · fc874df
1 parent 56d3092
commit fc874df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/caffe_converter/convert_symbol.py b/tools/caffe_converter/convert_symbol.py
@@ -200,7 +200,7 @@ def proto2symbol(proto_file):
     return ret, input_dim
 
 def main():
-    symbol_string, output_name = proto2script(sys.argv[1])
+    symbol_string, output_name, input_dim = proto2script(sys.argv[1])
     if len(sys.argv) > 2:
         with open(sys.argv[2], 'w') as fout:
             fout.write(symbol_string)