diff --git a/dinov2/run/train/train.py b/dinov2/run/train/train.py
index c2366e9bf..48f90dccd 100644
--- a/dinov2/run/train/train.py
+++ b/dinov2/run/train/train.py
@@ -7,6 +7,8 @@
 import os
 import sys
 
+sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../..')
+
 from dinov2.logging import setup_logging
 from dinov2.train import get_args_parser as get_train_args_parser
 from dinov2.run.submit import get_args_parser, submit_jobs