Initial impl of WIP packed vit (navit)

huggingface · Sep 13, 2023 · 6461405 · 6461405
1 parent 983310d
commit 6461405
Show file tree

Hide file tree

Showing 2 changed files with 860 additions and 0 deletions.
diff --git a/timm/models/__init__.py b/timm/models/__init__.py
@@ -67,6 +67,7 @@
 from .vision_transformer_hybrid import *
 from .vision_transformer_relpos import *
 from .vision_transformer_sam import *
+from .vision_transformer_packed import *
 from .volo import *
 from .vovnet import *
 from .xception import *