Is it safe to manually call `datamodule.setup()`?

goku · October 6, 2020, 7:38pm

DataModule’s setup is called before LightningModule’s setup.

Lightning-AI/lightning/blob/c510a7f90077140d60c47adf8e1e73638c2d1017/pytorch_lightning/trainer/trainer.py#L833-L841


      
          def call_setup_hook(self, model):
              # call setup after the ddp process has connected
              stage_name = 'test' if self.testing else 'fit'
              if self.datamodule is not None:
                  called = self.datamodule.has_setup_test if self.testing else self.datamodule.has_setup_fit
                  if not called:
                      self.datamodule.setup(stage_name)
              self.setup(stage_name)
              model.setup(stage_name)