diff --git a/timm/models/swin_transformer.py b/timm/models/swin_transformer.py index 2c87cf29..471b94e8 100644 --- a/timm/models/swin_transformer.py +++ b/timm/models/swin_transformer.py @@ -383,8 +383,8 @@ class SwinTransformerStage(nn.Module): """ Args: dim: Number of input channels. - input_resolution: Input resolution. out_dim: Number of output channels. + input_resolution: Input resolution. depth: Number of blocks. downsample: Downsample layer at the end of the layer. num_heads: Number of attention heads.