Add initials (CR) to model and file

pull/1150/head
Christoph Reich 3 years ago
parent 90dc74c450
commit f227b88831

@ -785,7 +785,7 @@ class SwinTransformerStage(nn.Module):
return output return output
class SwinTransformerV2(nn.Module): class SwinTransformerV2CR(nn.Module):
r""" Swin Transformer V2 r""" Swin Transformer V2
A PyTorch impl of : `Swin Transformer V2: Scaling Up Capacity and Resolution` - A PyTorch impl of : `Swin Transformer V2: Scaling Up Capacity and Resolution` -
https://arxiv.org/pdf/2111.09883 https://arxiv.org/pdf/2111.09883
@ -825,7 +825,7 @@ class SwinTransformerV2(nn.Module):
sequential_self_attention: bool = False, sequential_self_attention: bool = False,
use_deformable_block: bool = False) -> None: use_deformable_block: bool = False) -> None:
# Call super constructor # Call super constructor
super(SwinTransformerV2, self).__init__() super(SwinTransformerV2CR, self).__init__()
# Save parameters # Save parameters
self.patch_size: int = patch_size self.patch_size: int = patch_size
self.input_resolution: Tuple[int, int] = input_resolution self.input_resolution: Tuple[int, int] = input_resolution
Loading…
Cancel
Save