fix errors

gau-nernst · Jul 7, 2023 · 5925030 · 5925030
1 parent a764334
commit 5925030
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/vision_toolbox/backbones/vovnet.py b/vision_toolbox/backbones/vovnet.py
@@ -111,7 +111,7 @@ def __init__(
             [ConvNormAct(in_channels if i == 0 else mid_channels, mid_channels) for i in range(num_layers)]
         )
         concat_channels = in_channels + mid_channels * num_layers
-        self.out_conv = ConvNormAct(concat_channels, out_channels, kernel_size=1, padding=0)
+        self.out_conv = ConvNormAct(concat_channels, out_channels, 1)
 
         self.ese = ESEBlock(out_channels) if ese else None
         self.residual = residual and (in_channels == out_channels)
@@ -160,7 +160,7 @@ def __init__(
         in_c = stem_channels
         for n, stage_c, n_l, out_c in zip(num_blocks_list, stage_channels_list, num_layers_list, out_channels_list):
             stage = nn.Sequential()
-            stage.add_module("max_pool", nn.MaxPool2d(3, 2, padding=1))
+            stage.add_module("max_pool", nn.MaxPool2d(3, 2, 1))
             for i in range(n):
                 stage.add_module(
                     f"module_{i}",

diff --git a/vision_toolbox/components.py b/vision_toolbox/components.py
@@ -7,7 +7,7 @@
 from torchvision.ops import DeformConv2d
 
 
-__all__ = ["ConvNormAct", "SeparableConv2d", "DeformableConv2d", "ESEBlock", "SPPBlock"]
+__all__ = ["ConvNormAct", "SeparableConv2d", "DeformableConv2d", "SPPBlock"]
 
 
 class ConvNormAct(nn.Sequential):
@@ -22,8 +22,6 @@ def __init__(
         norm: str = "bn",
         act: str = "relu",
     ):
-        if kernel_size % 2 == 0:
-            raise ValueError("Even kernel is not supported")
         super().__init__()
         self.conv = nn.Conv2d(
             in_channels,