mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #2409 from adamjstewart/models/vgg 

VGG ConvMlp: fix layer defaults/types
fix_levit_load
Ross Wightman 2025-01-15 07:37:26 -08:00 committed by GitHub
parent ef7dec8e43 6d21eb0d37
commit 63b2de7a6c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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, kernel_size=7,
mlp_ratio=1.0, mlp_ratio=1.0,
drop_rate: float = 0.2, drop_rate: float = 0.2,
act_layer: Optional[Type[nn.Module]] = None, act_layer: Type[nn.Module] = nn.ReLU,
conv_layer: Optional[Type[nn.Module]] = None, conv_layer: Type[nn.Module] = nn.Conv2d,
): ):
super(ConvMlp, self).__init__() super(ConvMlp, self).__init__()
self.input_kernel_size = kernel_size self.input_kernel_size = kernel_size