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