Skip to content

Commit 653232c

Browse files
THasthikakaushikb11
authored andcommitted
Fixed missing arguments in lr_find call (Lightning-AI#6784)
There seem to be 3 arguments missing in the `lr_find` call in the tunining.py file.
1 parent 716ade5 commit 653232c

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
@@ -60,7 +60,13 @@ def tune(self, model, train_dataloader, val_dataloaders, datamodule):
6060

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

6571
def scale_batch_size(
6672
self,

0 commit comments

Comments
 (0)