summaryrefslogtreecommitdiff
path: root/training/run_experiment.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-05-02 14:10:53 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-05-02 14:10:53 +0200
commit1baeae6b414f71906bd1480d3ddc393ae878bd63 (patch)
tree288550c42a7dfea43e8464b00adfa7e47ef2bc5e /training/run_experiment.py
parent1d0977585f01c42e9f6280559a1a98037907a62e (diff)
Working on attention
Diffstat (limited to 'training/run_experiment.py')
-rw-r--r--training/run_experiment.py4
1 files changed, 2 insertions, 2 deletions
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__":