Merge pull request #2409 from adamjstewart/models/vgg

VGG ConvMlp: fix layer defaults/types
huggingface · Jan 15, 2025 · 63b2de7 · 63b2de7
2 parents ef7dec8 + 6d21eb0
commit 63b2de7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/timm/models/vgg.py b/timm/models/vgg.py
@@ -38,8 +38,8 @@ def __init__(
             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