From 2f19e8b863c54d16c1eb855bc89391063def15ce Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 2 Oct 2022 02:55:53 +0200 Subject: Add support for hydra 1.2, add metadata file --- training/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'training/main.py') 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 -- cgit v1.2.3-70-g09d2