Update metaformers.py

pull/1647/head
Fredo Guan 3 years ago
parent 4e682f1fda
commit ddfdb543dc

@ -231,7 +231,6 @@ class Downsampling(nn.Module):
pre_norm=None, post_norm=None, pre_permute = False):
super().__init__()
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,
stride=stride, padding=padding)
self.post_norm = post_norm(out_channels) if post_norm else nn.Identity()

Loading…
Cancel
Save