I am very happy to see your pytorch implementation of Probabilistic Unet. I am using it for a large-scale research, and later will cite your work in.
May I ask, I have been successfully training the model. But in the validation step, I don't see any script about it. For simplicity, may I ask how can we predict a single image? (is it the output of reconstruct() ? )
Dear stefanknegt,
I am very happy to see your pytorch implementation of Probabilistic Unet. I am using it for a large-scale research, and later will cite your work in.
May I ask, I have been successfully training the model. But in the validation step, I don't see any script about it. For simplicity, may I ask how can we predict a single image? (is it the output of reconstruct() ? )
Please help me. Thanks so much for your support.
Kind regards, Kha.