How to download the prediction file from the model developed with pytorch lightning?
I am trying to implement this,

https://github.com/bighuang624/DSANet
and also making a few changes to my code. I would like to know if you have any formats to extract the prediction file after training and validation?

PS. I am also following this documentation,

https://pytorch-lightning.readthedocs.io/_/downloads/en/latest/pdf/

Any help would be appreciated! @williamFalcon