diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-09-20 01:15:23 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-09-20 01:15:23 +0200 |
| commit | 2b743f2f9046a2e2930647d234dc7392d71efa66 (patch) | |
| tree | 237b5729cfbae90877a44baa52ac2ebaec388f79 /src/training/run_experiment.py | |
| parent | e181195a699d7fa237f256d90ab4dedffc03d405 (diff) | |
Fixed some bash scripts.
Diffstat (limited to 'src/training/run_experiment.py')
| -rw-r--r-- | src/training/run_experiment.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/training/run_experiment.py b/src/training/run_experiment.py index a347d9f..cc882ad 100644 --- a/src/training/run_experiment.py +++ b/src/training/run_experiment.py @@ -116,7 +116,7 @@ def load_modules_and_arguments(experiment_config: Dict) -> Tuple[Callable, Dict] # Learning rate scheduler lr_scheduler_ = None lr_scheduler_args = None - if experiment_config["lr_scheduler"] is not None: + if "lr_scheduler" in experiment_config: lr_scheduler_ = getattr( torch.optim.lr_scheduler, experiment_config["lr_scheduler"]["type"] ) |