Update README.md

hula-ai · Aug 24, 2021 · 64f9d84 · 64f9d84
1 parent 38aae44
commit 64f9d84
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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
 ```