lr_finder: Fix typo in docstring (#1746)

2020-05-06 18:39:22 +02:00 · 2020-05-06 18:39:22 +02:00 · b9364f96b1
parent 851866333c
commit b9364f96b1
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/trainer/lr_finder.py
+++ b/pytorch_lightning/trainer/lr_finder.py
@ -94,7 +94,7 @@ class TrainerLRFinderMixin(ABC):

            # Inspect results
            fig = lr_finder.plot(); fig.show()
-            suggested_lr = lr_finder.suggest()
+            suggested_lr = lr_finder.suggestion()

            # Overwrite lr and create new model
            hparams.lr = suggested_lr