| 1125 | self.val_check_batch = max(1, self.val_check_batch) |
| 1126 | |
| 1127 | def init_val_dataloader(self, model): |
| 1128 | self.get_val_dataloaders = model.val_dataloader |
| 1129 | self.num_val_batches = 0 |
| 1130 | if self.get_val_dataloaders() is not None: |
| 1131 | if isinstance(self.get_val_dataloaders()[0], torch.utils.data.DataLoader): |
| 1132 | self.num_val_batches = sum(len(dataloader) for dataloader in self.get_val_dataloaders()) |
| 1133 | self.num_val_batches = int(self.num_val_batches) |
| 1134 | else: |
| 1135 | self.num_val_batches = float('inf') |
| 1136 | |
| 1137 | def init_test_dataloader(self, model): |
| 1138 | self.get_test_dataloaders = model.test_dataloader |