update UNet

pme0 · Jul 1, 2023 · f3db607 · f3db607
1 parent 0b33c5b
commit f3db607
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deeplightning/model/unet.py b/deeplightning/model/unet.py
@@ -72,6 +72,7 @@ def forward(self, x, y):
         # shapes. Should we pad from right or left, top or bottom?
         x = F.pad(x, (0, 1, 0, 1))
         x = self.upconv(x)
+        cropped_y = self.crop(y, x)
         x = torch.cat((cropped_y, x), dim=1)
         x = self.conv(x)
         return x