Merge pull request #2409 from adamjstewart/models/vgg

VGG ConvMlp: fix layer defaults/types
2025-01-15 07:37:26 -08:00 · 2025-01-15 07:37:26 -08:00 · 63b2de7a6c
parent ef7dec8e43 6d21eb0d37
commit 63b2de7a6c
1 changed files with 2 additions and 2 deletions
--- a/timm/models/vgg.py
+++ b/timm/models/vgg.py
@ -38,8 +38,8 @@ class ConvMlp(nn.Module):
            kernel_size=7,
            mlp_ratio=1.0,
            drop_rate: float = 0.2,
-            act_layer: Optional[Type[nn.Module]] = None,
-            conv_layer: Optional[Type[nn.Module]] = None,
+            act_layer: Type[nn.Module] = nn.ReLU,
+            conv_layer: Type[nn.Module] = nn.Conv2d,
    ):
        super(ConvMlp, self).__init__()
        self.input_kernel_size = kernel_size