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 #