TinyImageNetDataModule¶
- class torch_uncertainty.datamodules.TinyImageNetDataModule(root, batch_size, eval_ood=False, eval_shift=False, shift_severity=1, val_split=None, ood_ds='svhn', interpolation='bilinear', basic_augment=True, 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]