From 32aab2f3b3c24faac80f7124522445f56f68c2a1 Mon Sep 17 00:00:00 2001
From: "s.islam" <s.islam@fz-juelich.de>
Date: Sat, 7 May 2022 16:11:33 +0200
Subject: [PATCH] REPORT-42: Ch:2, inceptionV4, RMI
---
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 00d545e..b5b6d61 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 = RMILoss(with_logits=True) #torch.nn.L1Loss() #RMILoss(with_logits=True) #torch.nn.MSELoss()
def forward(self, x):
x = self.model(x)
--
GitLab