search

JingZhang617 / UCNet

UC-Net: Uncertainty Inspired RGB-D Saliency Detectionvia Conditional Variational Autoencoders, CVPR2020
178 stars 25 forks source link

Training error "NotImplementedError" #6

Open simantika opened 3 years ago

simantika commented 3 years ago

Traceback (most recent call last): File "train.py", line 136, in pred_post, pred_prior, lattent_loss, depth_pred_post, depth_pred_prior = generator.forward(images,depths,gts) File "/home/paramshavakmahesha/Documents/TRY4/UCNet/model/ResNet_models.py", line 159, in forward lattent_loss = torch.mean(self.kl_divergence(self.posterior, self.prior)) File "/home/paramshavakmahesha/Documents/TRY4/UCNet/model/ResNet_models.py", line 146, in kl_divergence kl_div = kl.kl_divergence(posterior_latent_space, prior_latent_space) File "/home/paramshavakmahesha/anaconda3/envs/picanet_3.5/lib/python3.5/site-packages/torch/distributions/kl.py", line 161, in kl_divergence raise NotImplementedError NotImplementedError

ZhouCX117 commented 3 years ago

Maybe you can refer to https://github.com/pytorch/pytorch/issues/13545. I don't encounter this issue.