From 4fc6e51cf7b7085fddfb62e8ebe19cf6ac3f7e71 Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Tue, 16 Mar 2021 11:31:02 +0000 Subject: [PATCH] Add missing stage --- pytorch_lightning/trainer/connectors/data_connector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/connectors/data_connector.py b/pytorch_lightning/trainer/connectors/data_connector.py index 8b2899fc24b27..6bd6b7bec5068 100644 --- a/pytorch_lightning/trainer/connectors/data_connector.py +++ b/pytorch_lightning/trainer/connectors/data_connector.py @@ -81,7 +81,7 @@ def attach_data(self, model, train_dataloader, val_dataloaders, datamodule): # set up the passed in dataloaders (if needed) self.attach_dataloaders(model, train_dataloader, val_dataloaders) - self.attach_datamodule(model, datamodule) + self.attach_datamodule(model, datamodule, 'fit') self._validate_data_hooks(model) def __enforce_datamodule_dataloader_override(self, train_dataloader, val_dataloaders, datamodule):