(self)
| 37 | ) |
| 38 | |
| 39 | def val_dataloader(self): |
| 40 | return torch.utils.data.DataLoader( |
| 41 | self.dataset["test"], |
| 42 | batch_size=self.hparams.batch_size, |
| 43 | num_workers=self.hparams.num_workers, # NOTE: disable for debugging |
| 44 | drop_last=True, |
| 45 | collate_fn=self.hparams.collate_fn, |
| 46 | ) |
| 47 | |
| 48 | |
| 49 | class AstroClipCollator: |
no outgoing calls
no test coverage detected