From 10eafdf1640e955b5c0d7fd38133fab6bec626f5 Mon Sep 17 00:00:00 2001 From: "s.islam" <s.islam@fz-juelich.de> Date: Sun, 8 May 2022 02:41:41 +0200 Subject: [PATCH] REPORT-47: Ch:2, efficient_B4, MSE --- code/model_Unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/model_Unet.py b/code/model_Unet.py index 0e8c1dc..b16c188 100644 --- a/code/model_Unet.py +++ b/code/model_Unet.py @@ -37,7 +37,7 @@ class TestModule(pl.LightningModule): classes=1, # classes == output channels. We use one output channel for cyto data activation="sigmoid" ) - self.loss_f = torch.nn.MSELoss() #torch.nn.L1Loss() #RMILoss(with_logits=True) #torch.nn.MSELoss() + self.loss_f = torch.nn.MSELoss() #torch.nn.L1Loss() #torch.nn.MSELoss() #RMILoss(with_logits=True) def forward(self, x): x = self.model(x) -- GitLab