Create config script outputs C++ format vector

Author: serizba

examples/load_model/create_config_options.py

@@ -4,7 +4,7 @@ def create_serialized_options(fraction, growth):
     gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=fraction, allow_growth=growth)
     config = tf.ConfigProto(gpu_options=gpu_options)
     serialized = config.SerializeToString()
-    return list(map(hex, serialized))
+    return '{' + ','.join(list(map(hex, serialized))) + '}'
 
 if __name__ == "__main__":
     print("Create serialized options which allow TF to use a certain percentage of GPU memory and allow TF to expand this memory if required.")

examples/load_model/create_model.py

@@ -13,7 +13,7 @@ def example_1():
     i = tf.initializers.global_variables()
 
     # Write the model definition
-    with open('models/load_model.pb', 'wb') as f:
+    with open('load_model.pb', 'wb') as f:
         f.write(tf.get_default_graph().as_graph_def().SerializeToString())