Skip to content

Commit 17af914

Browse files
committed
Fix a bug for unused x/h effnets
1 parent 3f58765 commit 17af914

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/_efficientnet_blocks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def __init__(
150150
if s2d == 1:
151151
sd_chs = int(in_chs * 4)
152152
self.conv_s2d = create_conv2d(in_chs, sd_chs, kernel_size=2, stride=2, padding='same')
153-
self.bn_s2d = norm_act_layer(sd_chs, sd_chs)
153+
self.bn_s2d = norm_act_layer(sd_chs)
154154
dw_kernel_size = (dw_kernel_size + 1) // 2
155155
dw_pad_type = 'same' if dw_kernel_size == 2 else pad_type
156156
in_chs = sd_chs

0 commit comments

Comments
 (0)