Update S10.py

S10.py

@@ -115,7 +115,7 @@ class LtAutoEncoder(L.LighteningModule):
             train_acc.append(100*correct/processed)
         
         
-    def test_model(self, batch, batch_idx):
+    def testing_step(self, batch, batch_idx):
         test_losses = []
         test_acc = []
         model.eval()
@@ -146,10 +146,6 @@ class LtAutoEncoder(L.LighteningModule):
 model=LtAutoEncoder(Net)  
 
 
-# setup data
-dataset = MNIST(os.getcwd(), download=True, transform=ToTensor())
-train_loader = utils.data.DataLoader(dataset)
-
 # setup data
 dataset = CIFAR10(os.getcwd(), download=True, transform=ToTensor())
 train_loader = utils.data.DataLoader(dataset)