From f227b88831b55ba0bec754c92c2a279f2c5ecd3e Mon Sep 17 00:00:00 2001 From: Christoph Reich <34400551+ChristophReich1996@users.noreply.github.com> Date: Sat, 19 Feb 2022 22:14:38 +0100 Subject: [PATCH] Add initials (CR) to model and file --- .../{swin_transformer_v2.py => swin_transformer_v2_cr.py} | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) rename timm/models/{swin_transformer_v2.py => swin_transformer_v2_cr.py} (99%) diff --git a/timm/models/swin_transformer_v2.py b/timm/models/swin_transformer_v2_cr.py similarity index 99% rename from timm/models/swin_transformer_v2.py rename to timm/models/swin_transformer_v2_cr.py index c146ecc8..d7c36519 100644 --- a/timm/models/swin_transformer_v2.py +++ b/timm/models/swin_transformer_v2_cr.py @@ -785,7 +785,7 @@ class SwinTransformerStage(nn.Module): return output -class SwinTransformerV2(nn.Module): +class SwinTransformerV2CR(nn.Module): r""" Swin Transformer V2 A PyTorch impl of : `Swin Transformer V2: Scaling Up Capacity and Resolution` - https://arxiv.org/pdf/2111.09883 @@ -825,7 +825,7 @@ class SwinTransformerV2(nn.Module): sequential_self_attention: bool = False, use_deformable_block: bool = False) -> None: # Call super constructor - super(SwinTransformerV2, self).__init__() + super(SwinTransformerV2CR, self).__init__() # Save parameters self.patch_size: int = patch_size self.input_resolution: Tuple[int, int] = input_resolution