Fix typo

cvg · Oct 23, 2023 · dc6939e · dc6939e
1 parent e0104fd
commit dc6939e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gluefactory/train.py b/gluefactory/train.py
@@ -331,7 +331,7 @@ def sigint_handler(signal, frame):
 
     results = None  # fix bug with it saving
 
-    lr_scheduler = get_lr_scheduler(optimizer=optimizer, conf=conf.train.lr_scheduler)
+    lr_scheduler = get_lr_scheduler(optimizer=optimizer, conf=conf.train.lr_schedule)
     if args.restore:
         optimizer.load_state_dict(init_cp["optimizer"])
         if "lr_scheduler" in init_cp: