From 1baeae6b414f71906bd1480d3ddc393ae878bd63 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm 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/run_experiment.py') 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