Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 59fbcd8

Browse files
committedMar 21, 2023
[Feature] add model script of cait
1 parent aaaaec6 commit 59fbcd8

File tree

2 files changed

+438
-0
lines changed

2 files changed

+438
-0
lines changed
 

‎mindcv/models/__init__.py

+3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""models init"""
22
from . import (
33
bit,
4+
cait,
45
coat,
56
convit,
67
convnext,
@@ -49,6 +50,7 @@
4950
xcit,
5051
)
5152
from .bit import *
53+
from .cait import *
5254
from .coat import *
5355
from .convit import *
5456
from .convnext import *
@@ -101,6 +103,7 @@
101103
# we cannot use net.__all__, so we manually copy net.__all__ here.
102104
__all__ = []
103105
__all__.extend(bit.__all__)
106+
__all__.extend(cait.__all__)
104107
__all__.extend(coat.__all__)
105108
__all__.extend(convit.__all__)
106109
__all__.extend(convnext.__all__)

0 commit comments

Comments
 (0)