From 8e209cd86f3f27b4ab2a2e87f76953f9d52930f0 Mon Sep 17 00:00:00 2001 From: Kentaro Yoshioka Date: Thu, 23 Jun 2022 09:44:06 +0900 Subject: [PATCH] update recipe --- README.md | 22 ++++++++++------------ 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/README.md b/README.md index 4c5dcd3..9ab6999 100644 --- a/README.md +++ b/README.md @@ -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..