From 2b743f2f9046a2e2930647d234dc7392d71efa66 Mon Sep 17 00:00:00 2001 From: aktersnurra Date: Sun, 20 Sep 2020 01:15:23 +0200 Subject: Fixed some bash scripts. --- src/training/run_experiment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/training/run_experiment.py') 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"] ) -- cgit v1.2.3-70-g09d2