From 2bd043ce5d6ca079e1aed589e33a3721d46a99bb Mon Sep 17 00:00:00 2001 From: akiyuki ishikawa Date: Tue, 5 Dec 2023 11:13:17 +0900 Subject: [PATCH] fix doc position --- timm/models/swin_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/swin_transformer.py b/timm/models/swin_transformer.py index 2c87cf29bb..471b94e84e 100644 --- a/timm/models/swin_transformer.py +++ b/timm/models/swin_transformer.py @@ -383,8 +383,8 @@ def __init__( """ 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.