Update convert.py

caffe2fluid/convert.py

@@ -41,7 +41,11 @@ def convert(def_path, caffemodel_path, data_output_path, code_output_path,
                 np.save(data_out, data)
         if code_output_path:
             print_stderr('Saving source...')
+            s = sys.version
             with open(code_output_path, 'wb') as src_out:
+                if s.startswith('2'):
+                    src_out.write(transformer.transform_source())
+                else:
                     src_out.write(str.encode(transformer.transform_source()))
         print_stderr('set env variable before using converted model '\
                 'if used custom_layers:')