Update S10.py

S10.py

@@ -138,15 +138,15 @@ class DataModule(pl.LightningDataModule):
         ])
 
       
-    # prepare transforms standard to MNIST
-    self.mnist_train = CIFAR10(os.getcwd(), train=True, download=True, transform=train_transform)
-    self.mnist_test = CIFAR10(os.getcwd(), train=False, download=True, transform=valid_transform)
+    # prepare transforms standard to CIFAR10
+    self.cifar_train = CIFAR10(os.getcwd(), train=True, download=True, transform=train_transform)
+    self.cifar_test = CIFAR10(os.getcwd(), train=False, download=True, transform=valid_transform)
 
   def train_dataloader(self):
-    return DataLoader(self.mnist_train, batch_size=512)
+    return DataLoader(self.cifar_train, batch_size=512)
 
   def val_dataloader(self):
-    return DataLoader(self.mnist_test, batch_size=512)
+    return DataLoader(self.cifar_test, batch_size=512)
 
 data_module = DataModule()