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:52:13 -08:00
parent bc37dec29b
commit 4e682f1fda
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/metaformers.py
View File

@ -228,7 +228,7 @@ class Downsampling(nn.Module):
""" """
def __init__(self, in_channels, out_channels, def __init__(self, in_channels, out_channels,
kernel_size, stride=1, padding=0, kernel_size, stride=1, padding=0,
pre_norm=None, post_norm=None): 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.pre_permute = pre_permute