Update resnet.py

resnet.py

@@ -193,7 +193,7 @@ class custom_ResNet(pl.LightningModule):
 
         # Assign test dataset for use in dataloader(s)
         if stage == "test" or stage is None:
-            self.cifar_test = CIFAR10(self.data_dir, train=False, transform=self.test_transform)
+            self.cifar_test = CIFAR10(self.data_dir, train=False, download=True, transform=self.test_transform)
 
     def train_dataloader(self):
         return DataLoader(self.cifar_train, batch_size=BATCH_SIZE, num_workers=os.cpu_count())