(self)
| 55 | _ = self.test_dataset |
| 56 | |
| 57 | def train_dataloader(self): |
| 58 | dataloader_options = self.dataloader_options.copy() |
| 59 | dataloader_options["batch_size"] = self.cfg.TRAIN.BATCH_SIZE |
| 60 | dataloader_options["num_workers"] = self.cfg.TRAIN.NUM_WORKERS |
| 61 | return DataLoader( |
| 62 | self.train_dataset, |
| 63 | shuffle=False, |
| 64 | persistent_workers=True, |
| 65 | **dataloader_options, |
| 66 | ) |
| 67 | |
| 68 | def predict_dataloader(self): |
| 69 | dataloader_options = self.dataloader_options.copy() |