Removed extra loss functions

Author: Stefan

losses.py

@@ -3,7 +3,6 @@
 from unet_blocks import *
 from unet import Unet
 from utils import init_weights,init_weights_orthogonal_normal, l2_regularisation
-from losses import lovasz_hinge, dice_loss, binary_xloss
 import torch.nn.functional as F
 from torch.distributions import Normal, Independent, kl
 
@@ -277,7 +276,4 @@ def elbo(self, segm, analytic_kl=True, reconstruct_posterior_mean=False):
  self.reconstruction_loss = torch.sum(reconstruction_loss)
  self.mean_reconstruction_loss = torch.mean(reconstruction_loss)
 
- #self.reconstruction_loss = dice_loss(torch.sigmoid(self.reconstruction), segm)
- #self.reconstruction_loss = lovasz_hinge(self.reconstruction, segm)
-
  return -(self.reconstruction_loss + self.beta * self.kl)