Fix docstring.

cvg · Jan 24, 2024 · 5863345 · 5863345
1 parent 1c8d315
commit 5863345
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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