Skip to content

Commit

Permalink
update recipe
Browse files Browse the repository at this point in the history
  • Loading branch information
kentaroy47 authored Jun 23, 2022
1 parent 775548f commit 8e209cd
Showing 1 changed file with 10 additions and 12 deletions.
22 changes: 10 additions & 12 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,27 +19,25 @@ I use pytorch for implementation.
* Changed default training settings for ViT.

# Usage example
`python train_cifar10.py --lr 1e-4 --aug --n_epochs 400` # vit-patchsize-4
`python train_cifar10.py` # vit-patchsize-4

`python train_cifar10.py --lr 1e-4 --aug --n_epochs 200 --size 48` # vit-patchsize-4-imsize-48
`python train_cifar10.py --size 48` # vit-patchsize-4-imsize-48

`python train_cifar10.py --patch 2 --lr 1e-4 --aug --n_epochs 200` # vit-patchsize-2
`python train_cifar10.py --patch 2` # vit-patchsize-2

`python train_cifar10.py --net vit_small --lr 1e-4 --aug --n_epochs 400` # vit-small
`python train_cifar10.py --net vit_small --n_epochs 400` # vit-small

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

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

`python train_cifar10.py --net mlpmixer --lr 1e-4 --aug --n_epochs 400` # vit-small
`python train_cifar10.py --net mlpmixer --n_epochs 400` # vit-small

`python train_cifar10.py --net cait --lr 1e-4 --aug --n_epochs 200` # train with cait
`python train_cifar10.py --net cait --n_epochs 200` # train with cait

`python train_cifar10.py --net swin --lr 1e-4 --aug --n_epochs 400` # train with SwinTransformers
`python train_cifar10.py --net swin --n_epochs 400` # train with SwinTransformers

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

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

# Results..

Expand Down

0 comments on commit 8e209cd

Please sign in to comment.