Skip to content

Commit f581411

Browse files
authored
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.
1 parent 89b5326 commit f581411

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

pytorch_lightning/tuner/tuning.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,13 @@ def tune(self, model, train_dataloader, val_dataloaders, datamodule):
6161

6262
# Run learning rate finder:
6363
if self.trainer.auto_lr_find:
64-
self.lr_find(model, update_attr=True)
64+
self.lr_find(
65+
model,
66+
update_attr=True,
67+
train_dataloader=train_dataloader,
68+
val_dataloaders=val_dataloaders,
69+
datamodule=datamodule,
70+
)
6571

6672
self.trainer.state = TrainerState.FINISHED
6773

0 commit comments

Comments
 (0)