Skip to content

Commit 1d75f58

Browse files
committed
For completeness sake, pass pad_type to final UIB block in MSFA for MNV5, minor impact
1 parent 6239313 commit 1d75f58

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

timm/models/mobilenetv5.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def __init__(
4343
expansion_ratio: float = 2.0,
4444
interpolation_mode: str = "nearest",
4545
layer_scale_init_value: Optional[float] = None,
46+
pad_type: str = '',
4647
noskip: bool = True,
4748
act_layer: Optional[LayerType] = None,
4849
norm_layer: Optional[LayerType] = None,
@@ -62,6 +63,7 @@ def __init__(
6263
in_chs=self.in_channels,
6364
out_chs=self.out_channels,
6465
dw_kernel_size_mid=0,
66+
pad_type=pad_type,
6567
exp_ratio=self.expansion_ratio,
6668
act_layer=act_layer,
6769
norm_layer=norm_layer,
@@ -206,6 +208,7 @@ def __init__(
206208
in_chs=self.msfa_in_chs,
207209
out_chs=num_features,
208210
output_resolution=self.msfa_output_resolution,
211+
pad_type=pad_type,
209212
norm_layer=norm_layer,
210213
act_layer=act_layer,
211214
)
@@ -457,6 +460,7 @@ def __init__(
457460
in_chs=self.msfa_in_chs,
458461
out_chs=self.num_features,
459462
output_resolution=self.msfa_output_resolution,
463+
pad_type=pad_type,
460464
norm_layer=norm_layer,
461465
act_layer=act_layer,
462466
)

0 commit comments

Comments
 (0)