diff --git a/ML/Pytorch/object_detection/YOLO/train.py b/ML/Pytorch/object_detection/YOLO/train.py
index 6908e8f4..b87ec953 100755
--- a/ML/Pytorch/object_detection/YOLO/train.py
+++ b/ML/Pytorch/object_detection/YOLO/train.py
@@ -36,8 +36,8 @@
 PIN_MEMORY = True
 LOAD_MODEL = False
 LOAD_MODEL_FILE = "overfit.pth.tar"
-IMG_DIR = "data/images"
-LABEL_DIR = "data/labels"
+IMG_DIR = "data/data/images"
+LABEL_DIR = "data/data/labels"
 
 
 class Compose(object):