Skip to content

Commit 7f0ba4a

Browse files
authored
Add ConvMixer Models (#1281)
* initial * paper * config * ConvMixer Layer * Patch Embedding * ConvMixer * All Variants * Update __init__ * Update __init__ * Update black * tests * docs fix * apply * fix * fix * fix failing tests * Format * minor fix * update * Docs Fix
1 parent ff41c20 commit 7f0ba4a

File tree

3 files changed

+449
-0
lines changed

3 files changed

+449
-0
lines changed

keras_cv/models/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from keras_cv.models.convmixer import ConvMixer_512_16
16+
from keras_cv.models.convmixer import ConvMixer_768_32
17+
from keras_cv.models.convmixer import ConvMixer_1024_16
18+
from keras_cv.models.convmixer import ConvMixer_1536_20
19+
from keras_cv.models.convmixer import ConvMixer_1536_24
1520
from keras_cv.models.convnext import ConvNeXtBase
1621
from keras_cv.models.convnext import ConvNeXtLarge
1722
from keras_cv.models.convnext import ConvNeXtSmall

0 commit comments

Comments
 (0)