TinyImageNetDataModule¶
- class torch_uncertainty.datamodules.TinyImageNetDataModule(root, batch_size, eval_ood=False, val_split=None, ood_ds='svhn', rand_augment_opt=None, num_workers=1, pin_memory=True, persistent_workers=True)[source]¶
- test_dataloader()[source]¶
Get test dataloaders for TinyImageNet.
- Returns:
test set for in distribution data and out-of-distribution data.
- Return type:
List[DataLoader]