Add RepViT models

huggingface · Jul 21, 2023 · 3318e76 · 3318e76
1 parent 394e814
commit 3318e76
Show file tree

Hide file tree

Showing 2 changed files with 405 additions and 0 deletions.
diff --git a/timm/models/__init__.py b/timm/models/__init__.py
@@ -40,6 +40,7 @@
 from .pnasnet import *
 from .pvt_v2 import *
 from .regnet import *
+from .repvit import *
 from .res2net import *
 from .resnest import *
 from .resnet import *