Method __init__

pytorch_lightning/lightning_module.py:353–358

(self, config: Config, tokenizer)

# -----------------------------------------------------------------------------
class ELFDataModule(L.LightningDataModule):
    def __init__(self, config: Config, tokenizer):
        super().__init__()
        self.cfg = config
        self.tokenizer = tokenizer
        self._train_dataset = None
        self._eval_dataset = None

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