diff --git a/train.py b/train.py
index 8748677fd5f1..dc7c301ad031 100644
--- 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'