-
Notifications
You must be signed in to change notification settings - Fork 4
/
sample.sh
executable file
·29 lines (27 loc) · 1.2 KB
/
sample.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
# use z_var to change the variance of z for all the sampling
# use --mybn --accumulate_stats --num_standing_accumulations 32 to
# use running stats
python sample.py \
--experiment_name ImageNet2Places_l0.01_res0.1 \
--load_weights best2 \
--base_root ./ \
--dataset I128 --parallel --shuffle --num_workers 8 --batch_size 64 \
--n_class 5 --n_pretrain_classes 1000 \
--num_G_accumulations 8 --num_D_accumulations 8 \
--num_D_steps 1 --G_lr 1e-4 --D_lr 4e-4 --D_B2 0.999 --G_B2 0.999 \
--G_attn 64 --D_attn 64 \
--G_ch 96 --D_ch 96 \
--G_nl inplace_relu --D_nl inplace_relu \
--SN_eps 1e-6 --BN_eps 1e-5 --adam_eps 1e-6 \
--G_ortho 0.0 \
--G_shared \
--G_init ortho --D_init ortho --skip_init \
--hier --dim_z 120 --shared_dim 128 \
--use_multiepoch_sampler \
--test_every 2000 --save_every 1000 --num_best_copies 5 --num_save_copies 2 --seed 0 \
--skip_init --G_batch_size 512 --G_eval_mode --sample_trunc_curves 0.05_0.05_1.0 \
--sample_inception_metrics --sample_npz --sample_random --sample_sheets --sample_interps \
--sample_num_npz 50000 \
# Use if you also want to calculate KMMD when evaluating the model.
# In that case, "num_inception_images" is decreased to avoid a memory error
#--kmmd --num_inception_images 25000