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

fix doc position

Browse Source
This commit is contained in:
akiyuki ishikawa 2023-12-05 11:13:17 +09:00 committed by Ross Wightman
parent 4f2e1bf4cb
commit 2bd043ce5d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/swin_transformer.py
View File

@ -383,8 +383,8 @@ class SwinTransformerStage(nn.Module):
""" """
Args: Args:
dim: Number of input channels. dim: Number of input channels.
input_resolution: Input resolution.
out_dim: Number of output channels. out_dim: Number of output channels.
input_resolution: Input resolution.
depth: Number of blocks. depth: Number of blocks.
downsample: Downsample layer at the end of the layer. downsample: Downsample layer at the end of the layer.
num_heads: Number of attention heads. num_heads: Number of attention heads.