diff --git a/flash_examples/integrations/learn2learn/image_classification_imagenette_mini.py b/flash_examples/integrations/learn2learn/image_classification_imagenette_mini.py index 9fc4e82f34..1459acca63 100644 --- a/flash_examples/integrations/learn2learn/image_classification_imagenette_mini.py +++ b/flash_examples/integrations/learn2learn/image_classification_imagenette_mini.py @@ -33,7 +33,6 @@ # download MiniImagenet train_dataset = l2l.vision.datasets.MiniImagenet(root="data", mode="train", download=True) val_dataset = l2l.vision.datasets.MiniImagenet(root="data", mode="validation", download=True) -test_dataset = l2l.vision.datasets.MiniImagenet(root="data", mode="test", download=True) train_transform = { "to_tensor_transform": nn.Sequential( @@ -69,8 +68,6 @@ train_targets=torch.from_numpy(train_dataset.y.astype(int)), val_data=val_dataset.x, val_targets=torch.from_numpy(val_dataset.y.astype(int)), - test_data=test_dataset.x, - test_targets=torch.from_numpy(test_dataset.y.astype(int)), train_transform=train_transform, )