From 64f9d84c4787e47900689a03163e66164445c84a Mon Sep 17 00:00:00 2001 From: Saining Xie Date: Mon, 23 Aug 2021 18:12:04 -0700 Subject: [PATCH] Update README.md --- transfer/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/transfer/README.md b/transfer/README.md index b52351c..19b1755 100644 --- a/transfer/README.md +++ b/transfer/README.md @@ -101,7 +101,7 @@ python -u -m torch.distributed.launch --nproc_per_node=8 --use_env main.py \ ``` python -u -m torch.distributed.launch --nproc_per_node=8 --use_env main.py \ --batch-size 128 --output_dir [your output dir path] --epochs 100 --lr 3e-4 --weight-decay 0.1 --eval-freq 10 \ - --no-pin-mem --warmup-epochs 3 --data-set cifar10 --data-path [cifar-100 data path] --no-repeated-aug \ + --no-pin-mem --warmup-epochs 3 --data-set cifar100 --data-path [cifar-100 data path] --no-repeated-aug \ --resume [your pretrain checkpoint file] \ --reprob 0.0 --drop-path 0.1 --mixup 0.5 --cutmix 1 ``` @@ -110,7 +110,7 @@ python -u -m torch.distributed.launch --nproc_per_node=8 --use_env main.py \ ``` python -u -m torch.distributed.launch --nproc_per_node=8 --use_env main.py \ --batch-size 128 --output_dir [your output dir path] --epochs 100 --lr 3e-4 --weight-decay 0.3 --eval-freq 10 \ - --no-pin-mem --warmup-epochs 3 --data-set cifar10 --data-path [oxford-flowers data path] --no-repeated-aug \ + --no-pin-mem --warmup-epochs 3 --data-set flowers --data-path [oxford-flowers data path] --no-repeated-aug \ --resume [your pretrain checkpoint file] \ --reprob 0.25 --drop-path 0.1 --mixup 0 --cutmix 0 ``` @@ -119,7 +119,7 @@ python -u -m torch.distributed.launch --nproc_per_node=8 --use_env main.py \ ``` python -u -m torch.distributed.launch --nproc_per_node=8 --use_env main.py \ --batch-size 128 --output_dir [your output dir path] --epochs 100 --lr 3e-4 --weight-decay 0.1 --eval-freq 10 \ - --no-pin-mem --warmup-epochs 3 --data-set cifar10 --data-path [oxford-pets data path] --no-repeated-aug \ + --no-pin-mem --warmup-epochs 3 --data-set pets --data-path [oxford-pets data path] --no-repeated-aug \ --resume [your pretrain checkpoint file] \ --reprob 0 --drop-path 0 --mixup 0.8 --cutmix 0 ```