From 09d1ea628dab41011ebe9bf72f699e60ddf9d295 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Tue, 17 Jan 2023 10:58:07 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 679d7ed6..ce5e99d3 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -363,7 +363,7 @@ class RandomMixing(nn.Module): x = x.reshape(B, H, W, C) return x - +''' class LayerNormGeneral(nn.Module): r""" General LayerNorm for different situations. @@ -462,7 +462,7 @@ class LayerNormGeneral(nn.Module): if self.use_bias: x = x + self.bias return x -''' + class SepConv(nn.Module): r""" Inverted separable convolution from MobileNetV2: https://arxiv.org/abs/1801.04381.