Skip to content

Commit 6e01090

Browse files
committed
avg pool should not have been passed dd
1 parent 63d4dc9 commit 6e01090

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/byobnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def __init__(
285285
avg_stride = stride if dilation == 1 else 1
286286
if stride > 1 or dilation > 1:
287287
avg_pool_fn = AvgPool2dSame if avg_stride == 1 and dilation > 1 else nn.AvgPool2d
288-
self.pool = avg_pool_fn(2, avg_stride, ceil_mode=True, count_include_pad=False, **dd)
288+
self.pool = avg_pool_fn(2, avg_stride, ceil_mode=True, count_include_pad=False)
289289
else:
290290
self.pool = nn.Identity()
291291
self.conv = layers.conv_norm_act(in_chs, out_chs, 1, apply_act=apply_act, **dd)

0 commit comments

Comments
 (0)