From 7453639646158900fca7644f7bbc94cdbdf746fe Mon Sep 17 00:00:00 2001
From: Parcollet Titouan <titouan.parcollet@alumni.univ-avignon.fr>
Date: Thu, 22 Mar 2018 16:31:07 -0400
Subject: [PATCH] Cleaning

---
 scripts/training.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/training.py b/scripts/training.py
index df3e8b2..6d3c4a9 100644
--- a/scripts/training.py
+++ b/scripts/training.py
@@ -414,7 +414,9 @@ def train(d):
         quaternion = True            
     else:
         quaternion = False
-
+    
+    if not os.path.exists(d.workdir+"/LOGS"):
+        os.makedirs(d.workdir+"/LOGS")
     savedir = d.workdir+"/LOGS/"+d.save_prefix
 
     #
-- 
1.8.2.3