Merge branch 'main' of github.com:cvg/glue-factory into rpautrat/fix-…

…gluestick-config
cvg · Oct 30, 2023 · ae70916 · ae70916
2 parents 7e61e61 + ab6e6ee
commit ae70916
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: