From 1baeae6b414f71906bd1480d3ddc393ae878bd63 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 2 May 2021 14:10:53 +0200
Subject: Working on attention

---
 training/run_experiment.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'training')

diff --git a/training/run_experiment.py b/training/run_experiment.py
index 2b3ecab..4e045c7 100644
--- a/training/run_experiment.py
+++ b/training/run_experiment.py
@@ -149,9 +149,9 @@ def run(config: DictConfig) -> None:
 
 
 @hydra.main(config_path="conf", config_name="config")
-def main(cfg: DictConfig) -> None:
+def main(config: DictConfig) -> None:
     """Loads config with hydra."""
-    run(cfg)
+    run(config)
 
 
 if __name__ == "__main__":
-- 
cgit v1.2.3-70-g09d2