From 437ba4e22b2dad2ca309085a2e97f33dd03eb642 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 5 Jul 2021 00:21:43 +0200 Subject: Working on new hydra implementation --- training/main.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 training/main.py (limited to 'training/main.py') diff --git a/training/main.py b/training/main.py new file mode 100644 index 0000000..26d5aeb --- /dev/null +++ b/training/main.py @@ -0,0 +1,21 @@ +"""Loads config with hydra and runs experiment.""" +import hydra +from omegaconf import DictConfig + + +@hydra.main(config_path="conf", config_name="config") +def main(config: DictConfig) -> None: + """Loads config with hydra and runs the experiment.""" + import utils + from run import run + + utils.extras(config) + + if config.get("print_config"): + utils.print_config(config) + + return run(config) + + +if __name__ == "__main__": + main() -- cgit v1.2.3-70-g09d2