Commit d8360832 authored by Verena Praher's avatar Verena Praher
Browse files

add testing to the resnet

parent 0fa40e24
......@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment