Skip to content

Commit 63b2de7

Browse files
authored
Merge pull request #2409 from adamjstewart/models/vgg
VGG ConvMlp: fix layer defaults/types
2 parents ef7dec8 + 6d21eb0 commit 63b2de7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/vgg.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ def __init__(
3838
kernel_size=7,
3939
mlp_ratio=1.0,
4040
drop_rate: float = 0.2,
41-
act_layer: Optional[Type[nn.Module]] = None,
42-
conv_layer: Optional[Type[nn.Module]] = None,
41+
act_layer: Type[nn.Module] = nn.ReLU,
42+
conv_layer: Type[nn.Module] = nn.Conv2d,
4343
):
4444
super(ConvMlp, self).__init__()
4545
self.input_kernel_size = kernel_size

0 commit comments

Comments
 (0)