Update doclayout_yolo/nn/modules/g2l_crm.py

doclayout_yolo/nn/modules/g2l_crm.py

@@ -32,9 +32,17 @@ class DilatedBlock(nn.Module):
         self.dcv = Conv(c, c, k=self.k, s=1)
 
     def dilated_conv(self, x, dilation):
-        padding = dilation * (self.k // 2)
-        # Use conv2d with adjusted dilation and pass through full Conv module (includes act/bn)
-        return self.dcv(F.pad(x, (padding, padding, padding, padding), mode='constant', value=0), dilation=dilation)
+        act = self.dcv.act
+        bn = getattr(self.dcv, 'bn', None)  # Get bn if it exists, else None
+        weight = self.dcv.conv.weight
+        padding = dilation * (self.k//2)
+
+        out = F.conv2d(x, weight, stride=1, padding=padding, dilation=dilation)
+        if bn is not None:
+            out = bn(out)
+        out = act(out)
+        return out
+
 
     def forward(self, x):
         """'forward()' applies the YOLO FPN to input data."""