From 5863345b81cecdeef92b432b9acafe8a5e454f52 Mon Sep 17 00:00:00 2001
From: Alexander Veicht <veichta@ethz.ch>
Date: Wed, 24 Jan 2024 12:50:40 +0100
Subject: [PATCH] Fix docstring.

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

diff --git a/gluefactory/train.py b/gluefactory/train.py
index f885b597..ece43743 100644
--- a/gluefactory/train.py
+++ b/gluefactory/train.py
@@ -143,7 +143,7 @@ def filter_fn(x):
 
 
 def get_lr_scheduler(optimizer, conf):
-    """Get lr scheduler specified by conf."""
+    """Get lr scheduler specified by conf.train.lr_schedule."""
     if conf.type not in ["factor", "exp", None]:
         if hasattr(conf.options, "schedulers"):
             # Add option to chain multiple schedulers together