Skip to content

Commit fe2867c

Browse files
committed
Significant naflex refactor. Rename classes, models. Support flag for creating classic vits as naflex. Cleanup, improvements.
1 parent 72858c1 commit fe2867c

File tree

3 files changed

+501
-332
lines changed

3 files changed

+501
-332
lines changed

timm/models/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@
4141
from .mobilenetv3 import *
4242
from .mobilevit import *
4343
from .mvitv2 import *
44+
from .naflexvit import *
4445
from .nasnet import *
4546
from .nest import *
4647
from .nextvit import *
@@ -72,7 +73,6 @@
7273
from .visformer import *
7374
from .vision_transformer import *
7475
from .vision_transformer_hybrid import *
75-
from .vision_transformer_flex import *
7676
from .vision_transformer_relpos import *
7777
from .vision_transformer_sam import *
7878
from .vitamin import *

0 commit comments

Comments
 (0)