Merge pull request #51 from KPLauritzen/b-ensure-weights-dir-when-tra…

…ining Ensures weights/ dir exists
ultralytics · Jun 13, 2020 · 19e68e8 · 19e68e8
2 parents 099e6f5 + c3d4d32
commit 19e68e8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -20,6 +20,7 @@
     mixed_precision = False  # not installed
 
 wdir = 'weights' + os.sep  # weights dir
+os.makedirs(wdir, exist_ok=True)
 last = wdir + 'last.pt'
 best = wdir + 'best.pt'
 results_file = 'results.txt'