| 1135 | self.num_val_batches = float('inf') |
| 1136 | |
| 1137 | def init_test_dataloader(self, model): |
| 1138 | self.get_test_dataloaders = model.test_dataloader |
| 1139 | if self.get_test_dataloaders() is not None: |
| 1140 | if isinstance(self.get_test_dataloaders()[0], torch.utils.data.DataLoader): |
| 1141 | self.num_test_batches = sum(len(dataloader) for dataloader in self.get_test_dataloaders()) |
| 1142 | self.num_test_batches = int(self.num_test_batches) |
| 1143 | else: |
| 1144 | self.num_test_batches = float('inf') |
| 1145 | |
| 1146 | def evaluate(self, model, dataloaders, max_batches, test=False): |
| 1147 | """Run evaluation code. |