Skip to content

Releases: Rishit-dagli/ConvMixer-torch2tf

ConvMixer SavedModels

24 Oct 09:18
@Rishit-dagli Rishit-dagli
v0.1.0
37fd445
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.
Compare
Choose a tag to compare
ConvMixer SavedModels Latest
Latest

🔮This release includes:

  • TensorFlow SavedModels for 6 variants (more info below) of ConvMixer model pre-trained on ImageNet-1K
  • Code to convert the model to TensorFlow SavedModels
  • Example code to use these models for image classification

🚀Here are the models made public with this release:

Model Name Fine-tunable Resolution acc@1 #params File Size TensorFlow Model
ConvMixer-1536/20 ❌ 224x224 81.37 51.6 x 10^6 184MB github/drive/bucket
ConvMixer-768/32 ❌ 224x224 80.16 21.1 x 10^6 75MB github/drive/bucket
ConvMixer-1024/20 ❌ 224x224 76.94 24.4 x 10^6 87MB github/drive/bucket
Feature Extractor ConvMixer-1536/20 ✅ 224x224 - - 178MB github/drive/bucket
Feature Extractor ConvMixer-768/32 ✅ 224x224 - - 75MB github/drive/bucket
Feature Extractor ConvMixer-1024/20 ✅ 224x224 - - 75MB github/drive/bucket
Assets 8
Loading
0 Join discussion