gangogh: Generate 'generated' directory automatically

rodrigobdz · Nov 29, 2019 · 8d33721 · 8d33721
1 parent 504be83
commit 8d33721
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gangogh.py b/gangogh.py
@@ -1,4 +1,5 @@
 import os, sys
+from pathlib import Path
 
 sys.path.append(os.getcwd())
 
@@ -33,6 +34,8 @@
 PREITERATIONS = 2000  # Number of preiteration training cycles to run
 lib.print_model_settings(locals().copy())
 
+# Ensure that directory exists where ground truth and plots will be saved to.
+Path('generated').mkdir(parents=True, exist_ok=True)
 
 def GeneratorAndDiscriminator():
     return kACGANGenerator, kACGANDiscriminator