(self, config, tokenizer, dataset_reader)
| 7 | |
| 8 | class FinetuneDataModule(LightningDataModule): |
| 9 | def __init__(self, config, tokenizer, dataset_reader): |
| 10 | super().__init__() |
| 11 | self.config = config |
| 12 | self.tokenizer = tokenizer |
| 13 | self.dataset_reader = dataset_reader |
| 14 | |
| 15 | def prepare_data(self): |
| 16 | # download, split, etc... |