Commit 9ae2389

author

Stefan

committed

Removed extra loss functions

1 parent 3ad908b commit 9ae2389Copy full SHA for 9ae2389

File tree

-274

lines changed

-274

lines changed

Lines changed: 0 additions & 270 deletions

This file was deleted.

Lines changed: 0 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,6 @@`
`3`	`3`	`from unet_blocks import *`
`4`	`4`	`from unet import Unet`
`5`	`5`	`from utils import init_weights,init_weights_orthogonal_normal, l2_regularisation`
`6`		`-from losses import lovasz_hinge, dice_loss, binary_xloss`
`7`	`6`	`import torch.nn.functional as F`
`8`	`7`	`from torch.distributions import Normal, Independent, kl`
`9`	`8`
`@@ -277,7 +276,4 @@ def elbo(self, segm, analytic_kl=True, reconstruct_posterior_mean=False):`
`277`	`276`	`self.reconstruction_loss = torch.sum(reconstruction_loss)`
`278`	`277`	`self.mean_reconstruction_loss = torch.mean(reconstruction_loss)`
`279`	`278`
`280`		`- #self.reconstruction_loss = dice_loss(torch.sigmoid(self.reconstruction), segm)`
`281`		`- #self.reconstruction_loss = lovasz_hinge(self.reconstruction, segm)`
`282`		`-`
`283`	`279`	`return -(self.reconstruction_loss + self.beta * self.kl)`

Comments

(0)