Update S10.py

S10.py

@@ -61,7 +61,7 @@ class DataAugmentationTest():
 
     def __call__(self, image):
       image = np.array(image)
-      img = self.test_transform(image = image)['image']
+      image = self.test_transform(image = image)['image']
       return image
 
 
@@ -226,9 +226,5 @@ class Cifar10SearchDataset(pl.LightningModule):
         return DataLoader(self.cifar10_test, batch_size=64)
 
 
-model = Cifar10SearchDataset()
-trainer = pl.Trainer(accelerator="auto",max_epochs=1)
-#datasetcifar=Cifar10SearchDataset()
-#trainer = pl.Trainer()
-trainer.fit(model)
+