From a2ed92b1112ba9701417426a0e5a5625dbc43428 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miguel=20Gonz=C3=A1lez-Fierro?=
 <miguelgfierro@users.noreply.github.com>
Date: Wed, 7 Feb 2018 20:37:28 +0000
Subject: [PATCH] added os.makedirs in case log folder is not created

---
 run.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/run.py b/run.py
index 8ca723a..84c5775 100644
--- a/run.py
+++ b/run.py
@@ -13,6 +13,7 @@
 from logger import Logger
 from math import sqrt
 import numpy as np
+import os
 
 parser = argparse.ArgumentParser(description='RecoEncoder')
 parser.add_argument('--lr', type=float, default=0.00001, metavar='N',
@@ -126,7 +127,7 @@ def main():
                                is_constrained=args.constrained,
                                dp_drop_prob=args.drop_prob,
                                last_layer_activations=not args.skip_last_layer_nl)
-
+  os.makedirs(args.logdir, exist_ok=True)
   model_checkpoint = args.logdir + "/model"
   path_to_model = Path(model_checkpoint)
   if path_to_model.is_file():