From a0864ab7db6e68fd21a548361d80a333c36ab606 Mon Sep 17 00:00:00 2001 From: Tharindu Hasthika Date: Thu, 1 Apr 2021 17:42:08 +0530 Subject: [PATCH] Fixed missing arguments in `lr_find` call There seem to be 3 arguments missing in the `lr_find` call in the tunining.py file. --- pytorch_lightning/tuner/tuning.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/tuner/tuning.py b/pytorch_lightning/tuner/tuning.py index b9fa9afe0e77e..a7aa1ee256a5d 100644 --- 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