yjh0410 hace 2 años
padre
commit
dcd32839f6
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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