Computing model output over dataset

[goku]

self.write_prediction in LightningModule. It’s not documented yet.