Update tiny_vit.py to fix bug

nhathoang0110 · rwightman · commit 95ba90157fbb · 2023-09-23T10:05:52.000-07:00
diff --git a/timm/models/tiny_vit.py b/timm/models/tiny_vit.py
@@ -378,6 +378,7 @@ def __init__(
 
         super().__init__()
         self.depth = depth
+        self.out_dim =  out_dim
 
         # patch merging layer
         if downsample is not None:
