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

Update tiny_vit.py to fix bug

Browse Source
This commit is contained in:
Nguyen Nhat Hoang 2023-09-23 11:21:44 +07:00 committed by Ross Wightman
parent a9c29a05de
commit 95ba90157f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/tiny_vit.py
View File

@ -378,6 +378,7 @@ class TinyVitStage(nn.Module):
super().__init__() super().__init__()
self.depth = depth self.depth = depth
self.out_dim = out_dim
# patch merging layer # patch merging layer
if downsample is not None: if downsample is not None: