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

Add missing docs in SwinTransformerStage

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

1
timm/models/swin_transformer.py
View File

@ -384,6 +384,7 @@ class SwinTransformerStage(nn.Module):
Args: Args:
dim: Number of input channels. dim: Number of input channels.
input_resolution: Input resolution. input_resolution: Input resolution.
out_dim: Number of output channels.
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.

1
timm/models/swin_transformer_v2.py
View File

@ -378,6 +378,7 @@ class SwinTransformerV2Stage(nn.Module):
""" """
Args: Args:
dim: Number of input channels. dim: Number of input channels.
out_dim: Number of output channels.
input_resolution: Input resolution. input_resolution: Input resolution.
depth: Number of blocks. depth: Number of blocks.
num_heads: Number of attention heads. num_heads: Number of attention heads.