(self)
| 20 | self.num_workers = num_workers |
| 21 | |
| 22 | def train_dataloader(self): |
| 23 | return DataLoader( |
| 24 | dataset=self.train_set, |
| 25 | batch_size=self.batch_size, |
| 26 | shuffle=True, |
| 27 | pin_memory=self.pin_memory, |
| 28 | drop_last=False, |
| 29 | num_workers=self.num_workers, |
| 30 | persistent_workers=True |
| 31 | ) |
| 32 | |
| 33 | def val_dataloader(self): |
| 34 | return DataLoader( |