mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

VGG ConvMlp: fix layer defaults/types

Browse Source
This commit is contained in:
Adam J. Stewart 2025-01-15 12:11:56 +01:00
parent ef7dec8e43
commit 6d21eb0d37
No known key found for this signature in database
GPG Key ID: C66C0675661156FC
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/vgg.py
View File

@ -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