Merge branches 'master' and 'master' of https://github.com/himat/Buil…

…dingVision
himat · Apr 24, 2017 · 727039c · 727039c
2 parents 9f980eb + bb0583d
commit 727039c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/net.py b/net.py
@@ -140,6 +140,8 @@ def get_edges_file(f):
 D_solver = tf.train.AdamOptimizer().minimize(D_loss, var_list=theta_D)
 G_solver = tf.train.AdamOptimizer().minimize(G_loss, var_list=theta_G)
 
+if not os.path.exists('out/'):
+    os.makedirs('out/')
 
 iter_to_print = 5