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

Update metaformers.py

Browse Source
This commit is contained in:
Fredo Guan 2023-01-08 11:53:20 -08:00
parent 4e682f1fda
commit ddfdb543dc
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/metaformers.py
View File

@ -231,7 +231,6 @@ class Downsampling(nn.Module):
pre_norm=None, post_norm=None, pre_permute = False): pre_norm=None, post_norm=None, pre_permute = False):
super().__init__() super().__init__()
self.pre_norm = pre_norm(in_channels) if pre_norm else nn.Identity() self.pre_norm = pre_norm(in_channels) if pre_norm else nn.Identity()
self.pre_permute = pre_permute
self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=kernel_size, self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=kernel_size,
stride=stride, padding=padding) stride=stride, padding=padding)
self.post_norm = post_norm(out_channels) if post_norm else nn.Identity() self.post_norm = post_norm(out_channels) if post_norm else nn.Identity()