1 files changed, 2 insertions, 1 deletions
diff --git a/training/main.py b/training/main.py
index 26d5aeb..6e54a23 100644
--- a/training/main.py
+++ b/training/main.py
@@ -1,9 +1,10 @@
 """Loads config with hydra and runs experiment."""
 import hydra
 from omegaconf import DictConfig
+from training.metadata import TRAINING_DIR
 
 
-@hydra.main(config_path="conf", config_name="config")
+@hydra.main(version_base="1.2", config_path=TRAINING_DIR / "conf", config_name="config")
 def main(config: DictConfig) -> None:
     """Loads config with hydra and runs the experiment."""
     import utils