File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments