From dc6939e084f355790d616526e4145bd71f632352 Mon Sep 17 00:00:00 2001 From: Paul-Edouard Sarlin Date: Mon, 23 Oct 2023 21:36:30 +0200 Subject: [PATCH] Fix typo --- gluefactory/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gluefactory/train.py b/gluefactory/train.py index 12ad2075..debf2125 100644 --- 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: