Working on new hydra implementation

1 files changed, 21 insertions, 0 deletions

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()