Fixed missing arguments in lr_find call (#6784)

There seem to be 3 arguments missing in the `lr_find` call in the tunining.py file.
Lightning-AI · Apr 6, 2021 · f581411 · f581411
1 parent 89b5326
commit f581411
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pytorch_lightning/tuner/tuning.py b/pytorch_lightning/tuner/tuning.py
@@ -61,7 +61,13 @@ def tune(self, model, train_dataloader, val_dataloaders, datamodule):
 
         # Run learning rate finder:
         if self.trainer.auto_lr_find:
-            self.lr_find(model, update_attr=True)
+            self.lr_find(
+                model,
+                update_attr=True,
+                train_dataloader=train_dataloader,
+                val_dataloaders=val_dataloaders,
+                datamodule=datamodule,
+            )
 
         self.trainer.state = TrainerState.FINISHED