add testing to the resnet

models/resnet18.py

@@ -45,6 +45,14 @@ class Network(pl.LightningModule):
     def validation_end(self, outputs):
         return validation_end(outputs)
 
+    def test_step(self, data_batch, batch_nb):
+        return test_step(self, data_batch, batch_nb)
+
+    def test_end(self, outputs):
+        test_metrics = test_end(outputs)
+        self.experiment.log(test_metrics)
+        return test_metrics
+
     @pl.data_loader
     def tng_dataloader(self):
         return tng_dataloader()