Skip to content

Commit 073ffa8

Browse files
libertatislibertatis
committed
fix bug in all readme.md
fix bugs in cmdline args and model load args in all readme.md files
1 parent 8ff663d commit 073ffa8

File tree

34 files changed

+483
-483
lines changed

34 files changed

+483
-483
lines changed

‎README.md

+7-7
Original file line numberDiff line numberDiff line change
@@ -427,7 +427,7 @@ from visual_transformer import build_vit as build_model
427427
config = get_config('./configs/vit_base_patch16_224.yaml')
428428
# build model
429429
model = build_model(config)
430-
# load pretrained weights, .pdparams is NOT needed
430+
# load pretrained weights
431431
model_state_dict = paddle.load('./vit_base_patch16_224.pdparams')
432432
model.set_dict(model_state_dict)
433433
```
@@ -446,9 +446,9 @@ python main_single_gpu.py \
446446
-cfg=./configs/vit_base_patch16_224.yaml \
447447
-dataset=imagenet2012 \
448448
-batch_size=16 \
449-
-data_path=/dataset/imagenet \
449+
-data_path=/path/to/dataset/imagenet/val \
450450
-eval \
451-
-pretrained=./vit_base_patch16_224
451+
-pretrained=/path/to/pretrained/model/vit_base_patch16_224 # .pdparams is NOT needed
452452
```
453453

454454
<details>
@@ -468,9 +468,9 @@ python main_multi_gpu.py \
468468
-cfg=./configs/vit_base_patch16_224.yaml \
469469
-dataset=imagenet2012 \
470470
-batch_size=16 \
471-
-data_path=/dataset/imagenet \
471+
-data_path=/path/to/dataset/imagenet/val \
472472
-eval \
473-
-pretrained=./vit_base_patch16_224
473+
-pretrained=/path/to/pretrained/model/vit_base_patch16_224 # .pdparams is NOT needed
474474
```
475475

476476
</details>
@@ -488,7 +488,7 @@ python main_single_gpu.py \
488488
-cfg=./configs/vit_base_patch16_224.yaml \
489489
-dataset=imagenet2012 \
490490
-batch_size=32 \
491-
-data_path=/dataset/imagenet \
491+
-data_path=/path/to/dataset/imagenet/train
492492
```
493493

494494

@@ -509,7 +509,7 @@ python main_multi_gpu.py \
509509
-cfg=./configs/vit_base_patch16_224.yaml \
510510
-dataset=imagenet2012 \
511511
-batch_size=16 \
512-
-data_path=/dataset/imagenet \
512+
-data_path=/path/to/dataset/imagenet/train
513513
```
514514

515515
</details>

‎README_cn.md

+15-15
Original file line numberDiff line numberDiff line change
@@ -427,7 +427,7 @@ from visual_transformer import build_vit as build_model
427427
config = get_config('./configs/vit_base_patch16_224.yaml')
428428
# build model
429429
model = build_model(config)
430-
# load pretrained weights, .pdparams is NOT needed
430+
# load pretrained weights
431431
model_state_dict = paddle.load('./vit_base_patch16_224')
432432
model.set_dict(model_state_dict)
433433
```
@@ -443,12 +443,12 @@ or
443443
```shell
444444
CUDA_VISIBLE_DEVICES=0 \
445445
python main_single_gpu.py \
446-
-cfg='./configs/vit_base_patch16_224.yaml' \
447-
-dataset='imagenet2012' \
446+
-cfg=./configs/vit_base_patch16_224.yaml \
447+
-dataset=imagenet2012 \
448448
-batch_size=16 \
449-
-data_path='/dataset/imagenet' \
449+
-data_path=/path/to/dataset/imagenet/val \
450450
-eval \
451-
-pretrained='./vit_base_patch16_224'
451+
-pretrained=/path/to/pretrained/model/vit_base_patch16_224 # .pdparams is NOT needed
452452
```
453453

454454
<details>
@@ -465,12 +465,12 @@ or
465465
```shell
466466
CUDA_VISIBLE_DEVICES=0,1,2,3 \
467467
python main_multi_gpu.py \
468-
-cfg='./configs/vit_base_patch16_224.yaml' \
469-
-dataset='imagenet2012' \
468+
-cfg=./configs/vit_base_patch16_224.yaml \
469+
-dataset=imagenet2012 \
470470
-batch_size=16 \
471-
-data_path='/dataset/imagenet' \
471+
-data_path=/path/to/dataset/imagenet/val \
472472
-eval \
473-
-pretrained='./vit_base_patch16_224'
473+
-pretrained=/path/to/pretrained/model/vit_base_patch16_224 # .pdparams is NOT needed
474474
```
475475

476476
</details>
@@ -485,10 +485,10 @@ or
485485
```shell
486486
CUDA_VISIBLE_DEVICES=0 \
487487
python main_single_gpu.py \
488-
-cfg='./configs/vit_base_patch16_224.yaml' \
489-
-dataset='imagenet2012' \
488+
-cfg=./configs/vit_base_patch16_224.yaml \
489+
-dataset=imagenet2012 \
490490
-batch_size=32 \
491-
-data_path='/dataset/imagenet' \
491+
-data_path=/path/to/dataset/imagenet/train
492492
```
493493

494494

@@ -506,10 +506,10 @@ or
506506
```shell
507507
CUDA_VISIBLE_DEVICES=0,1,2,3 \
508508
python main_multi_gpu.py \
509-
-cfg='./configs/vit_base_patch16_224.yaml' \
510-
-dataset='imagenet2012' \
509+
-cfg=./configs/vit_base_patch16_224.yaml \
510+
-dataset=imagenet2012 \
511511
-batch_size=16 \
512-
-data_path='/dataset/imagenet' \
512+
-data_path=/path/to/dataset/imagenet/train
513513
```
514514

515515
</details>

‎gan/README.md

+5-5
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,8 @@ from generator import Generator
6666
config = get_config('./configs/styleformer_cifar10.yaml')
6767
# build model
6868
model = Generator(config)
69-
# load pretrained weights, .pdparams is NOT needed
70-
model_state_dict = paddle.load('./cifar10')
69+
# load pretrained weights
70+
model_state_dict = paddle.load('./cifar10.pdparams')
7171
model.set_dict(model_state_dict)
7272
```
7373

@@ -79,10 +79,10 @@ sh run_generate.sh
7979
or
8080
```shell
8181
python generate.py \
82-
-cfg='./configs/styleformer_cifar10.yaml' \
82+
-cfg=./configs/styleformer_cifar10.yaml \
8383
-num_out_images=16 \
84-
-out_folder='./images_cifar10' \
85-
-pretrained='./cifar10.pdparams'
84+
-out_folder=./images_cifar10 \
85+
-pretrained=/path/to/pretrained/model/cifar10 # .pdparams is NOT needed
8686
```
8787
The output images are stored in `-out_folder` path.
8888

‎gan/README_cn.md

+5-5
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@ from generator import Generator
6868
config = get_config('./configs/styleformer_cifar10.yaml')
6969
# build model
7070
model = Generator(config)
71-
# load pretrained weights, .pdparams is NOT needed
72-
model_state_dict = paddle.load('./cifar10')
71+
# load pretrained weights
72+
model_state_dict = paddle.load('./cifar10.pdparams')
7373
model.set_dict(model_state_dict)
7474
```
7575

@@ -81,10 +81,10 @@ sh run_generate.sh
8181
or
8282
```shell
8383
python generate.py \
84-
-cfg='./configs/styleformer_cifar10.yaml' \
84+
-cfg=./configs/styleformer_cifar10.yaml \
8585
-num_out_images=16 \
86-
-out_folder='./images_cifar10' \
87-
-pretrained='./cifar10.pdparams'
86+
-out_folder=./images_cifar10 \
87+
-pretrained=/path/to/pretrained/model/cifar10 # .pdparams is NOT needed
8888
```
8989
输出图像存储在 `-out_folder` 路径中.
9090

‎gan/Styleformer/README.md

+16-16
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ from generator import Generator
7272
config = get_config('./configs/styleformer_cifar10.yaml')
7373
# build model
7474
model = Generator(config)
75-
# load pretrained weights, .pdparams is NOT needed
76-
model_state_dict = paddle.load('./cifar10')
75+
# load pretrained weights
76+
model_state_dict = paddle.load('./cifar10.pdparams')
7777
model.set_dict(model_state_dict)
7878
```
7979

@@ -85,10 +85,10 @@ sh run_generate.sh
8585
or
8686
```shell
8787
python generate.py \
88-
-cfg='./configs/styleformer_cifar10.yaml' \
88+
-cfg=./configs/styleformer_cifar10.yaml \
8989
-num_out_images=16 \
90-
-out_folder='./images_cifar10' \
91-
-pretrained='./cifar10.pdparams'
90+
-out_folder=./images_cifar10 \
91+
-pretrained=/path/to/pretrained/model/cifar10 # .pdparams is NOT needed
9292
```
9393
The output images are stored in `-out_folder` path.
9494

@@ -102,11 +102,11 @@ or
102102
```shell
103103
CUDA_VISIBLE_DEVICES=0 \
104104
python main_single_gpu.py \
105-
-cfg='./configs/styleformer_cifar10.yaml' \
106-
-dataset='cifar10' \
105+
-cfg=./configs/styleformer_cifar10.yaml \
106+
-dataset=cifar10 \
107107
-batch_size=32 \
108108
-eval \
109-
-pretrained='./cifar10'
109+
-pretrained=/path/to/pretrained/model/cifar10 # .pdparams is NOT needed
110110
```
111111

112112
<details>
@@ -123,11 +123,11 @@ or
123123
```shell
124124
CUDA_VISIBLE_DEVICES=0,1,2,3 \
125125
python main_single_gpu.py \
126-
-cfg='./configs/styleformer_cifar10.yaml' \
127-
-dataset='cifar10' \
126+
-cfg=./configs/styleformer_cifar10.yaml \
127+
-dataset=cifar10 \
128128
-batch_size=32 \
129129
-eval \
130-
-pretrained='./cifar10'
130+
-pretrained=/path/to/pretrained/model/cifar10.pdparams # .pdparams is NOT needed
131131
```
132132

133133
</details>
@@ -143,9 +143,9 @@ or
143143
CUDA_VISIBLE_DEVICES=0 \
144144
python main_single_gpu.py \
145145
-cfg='./configs/styleformer_cifar10.yaml' \
146-
-dataset='cifar10' \
146+
-dataset=cifar10 \
147147
-batch_size=32 \
148-
-pretrained='./cifar10'
148+
-pretrained=/path/to/pretrained/model/cifar10 # .pdparams is NOT needed
149149
```
150150

151151
<details>
@@ -162,10 +162,10 @@ or
162162
```shell
163163
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
164164
python main_single_gpu.py \
165-
-cfg='./configs/styleformer_cifar10.yaml' \
166-
-dataset='cifar10' \
165+
-cfg=./configs/styleformer_cifar10.yaml \
166+
-dataset=cifar10 \
167167
-batch_size=32 \
168-
-pretrained='./cifar10'
168+
-pretrained=/path/to/pretrained/model/cifar10 # .pdparams is NOT needed
169169
```
170170

171171
</details>

‎gan/transGAN/readme.md

+17-17
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ from models.ViT_custom import Generator
4646
config = get_config('./configs/transgan_cifar10.yaml')
4747
# build model
4848
model = Generator(config)
49-
# load pretrained weights, .pdparams is NOT needed
50-
model_state_dict = paddle.load('./transgan_cifar10')
49+
# load pretrained weights
50+
model_state_dict = paddle.load('./transgan_cifar10.pdparams')
5151
model.set_dict(model_state_dict)
5252
```
5353

@@ -59,10 +59,10 @@ sh run_generate.sh
5959
or
6060
```shell
6161
python generate.py \
62-
-cfg='./configs/transgan_cifar10.yaml' \
62+
-cfg=./configs/transgan_cifar10.yaml \
6363
-num_out_images=16 \
64-
-out_folder='./images_cifar10' \
65-
-pretrained='./transgan_cifar10.pdparams'
64+
-out_folder=./images_cifar10 \
65+
-pretrained=/path/to/pretrained/model/transgan_cifar10 # .pdparams is NOT needed
6666
```
6767
The output images are stored in `-out_folder` path.
6868

@@ -76,11 +76,11 @@ or
7676
```shell
7777
CUDA_VISIBLE_DEVICES=0 \
7878
python main_single_gpu.py \
79-
-cfg="./configs/transgan_cifar10.yaml" \
80-
-dataset='cifar10' \
79+
-cfg=./configs/transgan_cifar10.yaml \
80+
-dataset=cifar10 \
8181
-batch_size=32 \
8282
-eval \
83-
-pretrained='./transgan_cifar10'
83+
-pretrained=/path/to/pretrained/model/transgan_cifar10.pdparams # .pdparams is NOT needed
8484
```
8585
<details>
8686

@@ -96,12 +96,12 @@ or
9696
```shell
9797
CUDA_VISIBLE_DEVICES=0,1,2,3 \
9898
python main_multi_gpu.py \
99-
-cfg='./configs/transgan_cifar10.yaml' \
100-
-dataset='cifar10' \
99+
-cfg=./configs/transgan_cifar10.yaml \
100+
-dataset=cifar10 \
101101
-batch_size=32 \
102-
-data_path='/dataset/imagenet' \
102+
-data_path=/path/to/dataset/imagenet/val \
103103
-eval \
104-
-pretrained='./transgan_cifar10'
104+
-pretrained=/path/to/pretrained/model/transgan_cifar10 # .pdparams is NOT needed
105105
```
106106

107107
</details>
@@ -117,8 +117,8 @@ or
117117
```shell
118118
CUDA_VISIBLE_DEVICES=0 \
119119
python main_single_gpu.py \
120-
-cfg="./configs/transgan_cifar10.yaml" \
121-
-dataset='cifar10' \
120+
-cfg=./configs/transgan_cifar10.yaml \
121+
-dataset=cifar10 \
122122
-batch_size=32 \
123123
```
124124
<details>
@@ -135,10 +135,10 @@ or
135135
```shell
136136
CUDA_VISIBLE_DEVICES=0,1,2,3 \
137137
python main_multi_gpu.py \
138-
-cfg='./configs/transgan_cifar10.yaml' \
139-
-dataset='cifar10' \
138+
-cfg=./configs/transgan_cifar10.yaml \
139+
-dataset=cifar10 \
140140
-batch_size=16 \
141-
-data_path='/dataset/imagenet' \
141+
-data_path=/path/to/dataset/imagenet/train
142142
```
143143

144144
</details>

0 commit comments

Comments
 (0)