Reduce memory footprint

Lightning-Universe · Nov 23, 2021 · 096f7fd · 096f7fd
1 parent 654d9b6
commit 096f7fd
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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,
 )