From a21fa20d7af6ebb6e940dfcb5a32087f5e2db9d8 Mon Sep 17 00:00:00 2001
From: "s.islam" <s.islam@fz-juelich.de>
Date: Fri, 6 May 2022 05:56:29 +0200
Subject: [PATCH] REPORT-13: Ch:1, densenet161, L1
---
code/model_Unet.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/code/model_Unet.py b/code/model_Unet.py
index 7f20b0a..f2e2186 100644
--- a/code/model_Unet.py
+++ b/code/model_Unet.py
@@ -31,13 +31,13 @@ class TestModule(pl.LightningModule):
# Define the model
self.model = smp.Unet(
- encoder_name="densenet121", # Also consider using smaller or larger encoders
+ encoder_name="densenet161", # Also consider using smaller or larger encoders
encoder_weights= "imagenet", # Do the pretrained weights help? Try with or without
in_channels=1, # We use 1 chanel transmittance as input
classes=1, # classes == output channels. We use one output channel for cyto data
activation="sigmoid"
)
- self.loss_f = RMILoss(with_logits=True) #torch.nn.L1Loss() #RMILoss(with_logits=True) #torch.nn.MSELoss()
+ self.loss_f = torch.nn.L1Loss() #torch.nn.L1Loss() #RMILoss(with_logits=True) #torch.nn.MSELoss()
def forward(self, x):
x = self.model(x)
--
GitLab