From ab6e6eef8b78f3293b0ef8cfa6c016f61ea649be Mon Sep 17 00:00:00 2001
From: Paul-Edouard Sarlin <15985472+sarlinpe@users.noreply.github.com>
Date: Mon, 23 Oct 2023 21:51:26 +0200
Subject: [PATCH] Fix typo (#32)

---
 gluefactory/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gluefactory/train.py b/gluefactory/train.py
index 12ad207..debf212 100644
--- a/gluefactory/train.py
+++ b/gluefactory/train.py
@@ -331,7 +331,7 @@ def training(rank, conf, output_dir, args):
 
     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: