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

Add initials (CR) to model and file

Browse Source
This commit is contained in:
Christoph Reich 2022-02-19 22:14:38 +01:00
parent 90dc74c450
commit f227b88831
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/swin_transformer_v2.py → timm/models/swin_transformer_v2_cr.py
View File

@ -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