Give PyTorch Lightning dummy data loaders so it runs the training loop. The ModelWrapper already has the training data from its __init__.
| 18 | |
| 19 | |
| 20 | class DataModuleOverfit(LightningDataModule): |
| 21 | """Give PyTorch Lightning dummy data loaders so it runs the training loop. The |
| 22 | ModelWrapper already has the training data from its __init__. |
| 23 | """ |
| 24 | |
| 25 | def train_dataloader(self): |
| 26 | return DataLoader(DummyDataset(), 1, num_workers=0) |
| 27 | |
| 28 | def val_dataloader(self): |
| 29 | return DataLoader(DummyDataset(limit=1), 1, num_workers=0) |