Update model/modnet.py

model/modnet.py

@@ -93,7 +93,7 @@ class LRBranch(nn.Module):
         self.backbone = backbone
         print("🧠 DEBUG enc_channels:", enc_channels)
         print("👉 len(enc_channels):", len(enc_channels))
-        self.se_block = SEBlock(enc_channels[4], enc_channels[4], reduction=4)
+        self.se_block = SEBlock(enc_channels[3], enc_channels[3], reduction=4)
         self.conv_lr16x = Conv2dIBNormRelu(enc_channels[4], enc_channels[3], 5, stride=1, padding=2)
         self.conv_lr8x = Conv2dIBNormRelu(enc_channels[3], enc_channels[2], 5, stride=1, padding=2)
         self.conv_lr = Conv2dIBNormRelu(enc_channels[2], 1, kernel_size=3, stride=2, padding=1, with_ibn=False, with_relu=False)