(self, hparams)
| 14 | |
| 15 | class DataModule(LightningDataModule): |
| 16 | def __init__(self, hparams): |
| 17 | super(DataModule, self).__init__() |
| 18 | self.hparams.update(hparams.__dict__) if hasattr(hparams, "__dict__") else self.hparams.update(hparams) |
| 19 | self._mean, self._std = None, None |
| 20 | self._saved_dataloaders = dict() |
| 21 | self.dataset = None |
| 22 | |
| 23 | def prepare_dataset(self): |
| 24 |