Method setup

pytorch_lightning/lightning_module.py:360–362

(self, stage=None)

        self._eval_dataset = None

    def setup(self, stage=None):
        if self._train_dataset is None:
            self._train_dataset, self._eval_dataset = load_dataset(self.cfg)

    def train_dataloader(self):
        # Lightning attaches the DistributedSampler when strategy=ddp.