Skip to content

About unofficial Implementation of MLP-Mixer, Conv-Mixer, ResMLP and Vit in PyTorch

Notifications You must be signed in to change notification settings

mskimS2/image-models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Model Zoo

This repository is an unofficial PyTorch implementation of Feedforward and Transformer based Networks (such as mlp-mixer, vit, ...). most experiments are performed on small benchmark datasets.

Experiemental Results

Model Params (M) CIFAR10 CIFAR100 SVHN
MLP-Mixer 1.13 84.73% 56.69% 93.88%
ResMLP 1.09 88.12% 64.72% 95.98%
Conv-Mixer 29.54 82.41% 57.19% 93.43%
Vit 18.94 83.41% 56.10% 94.84%

About

About unofficial Implementation of MLP-Mixer, Conv-Mixer, ResMLP and Vit in PyTorch

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published