forked from NVlabs/stylegan2-ada-pytorch
-
Notifications
You must be signed in to change notification settings - Fork 3
/
commands.sh
27 lines (14 loc) · 1.45 KB
/
commands.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#pretrained ffhq and ahqcats
python stylegan_blending.py --network1 pretrained/ffhq-res512-mirror-stylegan2-noaug.pkl --network2 network-snapshot-000080.pkl --outdir out_blend
# 256
python stylegan_blending.py --network1 pretrained/ffhq-res256-mirror-paper256-noaug.pkl --network2 00004-afhqcat256-mirror-auto2-resumeffhq256/network-snapshot-000560.pkl --outdir out_custom --dim 256
# dog
python stylegan_blending.py --network1 pretrained/ffhq-res256-mirror-paper256-noaug.pkl --network2 00007-afhqdog256-mirror-auto1-resumeffhq256/network-snapshot-000400.pkl --outdir out_custom_dog --dim 256
# wild
python stylegan_blending.py --network1 pretrained/ffhq-res256-mirror-paper256-noaug.pkl --network2 00009-afhqwild256-mirror-auto1-resumeffhq256/network-snapshot-000400.pkl --outdir out_custom_wild --dim 256
# cartoon
python stylegan_blending.py --network1 pretrained/ffhq-res256-mirror-paper256-noaug.pkl --network2 00008-cartoon256-mirror-auto1-resumeffhq256/network-snapshot-000400.pkl --outdir out_custom_cartoon --dim 256
# Training
CUDA_VISIBLE_DEVICES=0 python train.py --outdir ./training_runs --data ./datasets/afhqdog256.zip --gpus 1 --mirror 1 --resume=ffhq256 --snap=10
CUDA_VISIBLE_DEVICES=3 python train.py --outdir ./training_runs --data ./datasets/cartoon256.zip --gpus 1 --mirror 1 --resume=ffhq256 --snap=10
CUDA_VISIBLE_DEVICES=1 python train.py --outdir ./training_runs --data ./datasets/afhqwild256.zip --gpus 1 --mirror 1 --resume=ffhq256 --snap=10