def test(self):
    super().test()
    self.val_images.append(self.input)
    self.val_predictions.append(self.output)
    self.val_labels.append(self.label)
def test(self):
    super().test()
    self.val_images.append(self.input)
    self.val_predictions.append(self.output)
    self.val_labels.append(self.label)