Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
kentaroy47 authored Oct 11, 2021
1 parent 4bf98a9 commit c51293b
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,14 @@ Added [ConvMixer]((https://openreview.net/forum?id=TVHS5Y4dNvM)) implementation.

`python train_cifar10.py --patch 2` # vit-patchsize-2

`python train_cifar10.py --net res18` # resnet18

`python train_cifar10.py --net vit_timm --lr 1e-4` # train with pretrained vit

`python train_cifar10.py --net convmixer --aug --n_epochs 200` # train with convmixer

`python train_cifar10.py --net res18` # resnet18

`python train_cifar10.py --net res18 --aug --n_epochs 200` # resnet18+randaug

# Results..

| | Accuracy |
Expand All @@ -31,6 +33,7 @@ Added [ConvMixer]((https://openreview.net/forum?id=TVHS5Y4dNvM)) implementation.
| ViT base (timm transfer) | 98.5% |
| [ConvMixerTiny(no pretrain)](https://openreview.net/forum?id=TVHS5Y4dNvM) | 96.3% |
| resnet18 | 93% |
| resnet18+randaug | 95% |

# Used in..
* Vision Transformer Pruning [arxiv](https://arxiv.org/abs/2104.08500) [github](https://github.com/Cydia2018/ViT-cifar10-pruning)

0 comments on commit c51293b

Please sign in to comment.