fix doc position

2023-12-05 11:13:17 +09:00 · 2023-12-05 11:13:17 +09:00 · 2bd043ce5d
parent 4f2e1bf4cb
commit 2bd043ce5d
1 changed files with 1 additions and 1 deletions
--- 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.