yjh0410 2 anni fa
parent
commit
dcd32839f6
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      models/detectors/lodet/lodet_backbone.py

+ 1 - 1
models/detectors/lodet/lodet_backbone.py

@@ -21,7 +21,7 @@ class ScaleModulationNet(nn.Module):
         # P1/2
         self.layer_1 = nn.Sequential(
             Conv(3, 16, k=3, p=1, s=2, act_type=act_type, norm_type=norm_type),
-            Conv(16, 16, k=3, p=1, act_type=act_type, norm_type=norm_type, depthwise=depthwise),
+            Conv(16, 16, k=3, p=1, s=1, act_type=act_type, norm_type=norm_type, depthwise=depthwise),
         )
 
         # P2/4