-
Notifications
You must be signed in to change notification settings - Fork 2
/
run_gcpert.sh
29 lines (25 loc) · 1.03 KB
/
run_gcpert.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
25
26
27
28
29
#! /bin/bash
source /ocean/projects/asc170022p/yanwuxu/miniconda/etc/profile.d/conda.sh
conda activate cyclegan
pert_threshold=2.0 ###scale of perturbation
lambda_blank=50.0 ###constraint coefficient of perturbation
identity=10.0
direction='AtoB'
dataroot='./data/selfie2anime'
batch_size=4
load_size=144
crop_size=128
model=maxgcpert3_gan
bounded=unbounded
netG=resnet_6blocks
netD=basic
python train.py --dataroot $dataroot --model $model --gan_mode lsgan \
--bounded $bounded --grid_size 2 --pert_threshold $pert_threshold --lambda_blank $lambda_blank \
--pool_size 50 --no_dropout --load_size $load_size --crop_size $crop_size \
--netG $netG --netD $netD --batch_size $batch_size --identity $identity \
--direction $direction;
#python test.py --dataroot $dataroot --model $model --eval \
#--bounded $bounded --grid_size 2 --pert_threshold $pert_threshold --lambda_blank $lambda_blank \
#--no_dropout --load_size $load_size --crop_size $crop_size \
#--netG $netG --netD $netD --batch_size $batch_size --identity $identity \
#--direction $direction