diff --git a/pytorch_lightning/trainer/lr_finder.py b/pytorch_lightning/trainer/lr_finder.py index e664fd3cc4..5e8c7a862a 100755 --- 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