From 6d21eb0d37e943154418d9f54f46cc40d3a7ac55 Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Wed, 15 Jan 2025 12:11:56 +0100
Subject: [PATCH] VGG ConvMlp: fix layer defaults/types

---
 timm/models/vgg.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/timm/models/vgg.py b/timm/models/vgg.py
index c096df23f..a4cfbffdf 100644
--- 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