A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
270
stars
67
forks
source link
Hi, I'm getting an error of 'ValueError: The parameter loc has invalid values' while training the model with input channels as 1 and latent dim=2. #20
Closed
yashgarg98 closed 3 years ago