(self)
| 53 | ) |
| 54 | |
| 55 | def val_dataloader(self): |
| 56 | return torch.utils.data.DataLoader( |
| 57 | self.dev_dataset, |
| 58 | batch_size=self.config.eval_batch_size, |
| 59 | shuffle=False, |
| 60 | collate_fn=create_collate_fn(self.tokenizer.pad_token_id, pretrain=False), |
| 61 | num_workers=min([self.config.eval_batch_size, self.config.num_workers]), |
| 62 | ) |
| 63 | |
| 64 | def test_dataloader(self): |
| 65 | return torch.utils.data.DataLoader( |
nothing calls this directly
no test coverage detected