We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents ef7dec8 + 6d21eb0 commit 63b2de7Copy full SHA for 63b2de7
timm/models/vgg.py
@@ -38,8 +38,8 @@ def __init__(
38
kernel_size=7,
39
mlp_ratio=1.0,
40
drop_rate: float = 0.2,
41
- act_layer: Optional[Type[nn.Module]] = None,
42
- conv_layer: Optional[Type[nn.Module]] = None,
+ act_layer: Type[nn.Module] = nn.ReLU,
+ conv_layer: Type[nn.Module] = nn.Conv2d,
43
):
44
super(ConvMlp, self).__init__()
45
self.input_kernel_size = kernel_size
0 commit comments