diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-05-02 14:10:53 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-05-02 14:10:53 +0200 |
| commit | 1baeae6b414f71906bd1480d3ddc393ae878bd63 (patch) | |
| tree | 288550c42a7dfea43e8464b00adfa7e47ef2bc5e /training | |
| parent | 1d0977585f01c42e9f6280559a1a98037907a62e (diff) | |
Working on attention
Diffstat (limited to 'training')
| -rw-r--r-- | training/run_experiment.py | 4 |
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__": |