Sampling a Segmentation

[anwai98]

Hi Stefan, thanks for this nice PyTorch implementation. I have a question regarding the sample function in the probabilistic_unet.py file. In line 222 (while training), wouldn't we want to sample from the posterior net? (according to the paper's training heuristic, correct me if I am wrong)