File size: 1,555 Bytes
8c212a5 |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
#!/bin/bash
# =================================== #
# Experiment configuration #
# =================================== #
# ======== GAN Type / Corpus ======== #
gan="stylegan2_ffhq1024"
stylegan_space="W+"
stylegan_layer=11
corpus="complex"
# ==== Latent Support Sets (LSS) ==== #
num_latent_support_dipoles=64
min_shift_magnitude=0.1
max_shift_magnitude=0.2
lss_beta=0.5
# ==== Corpus Support Sets (CSS) ==== #
linear=false
styleclip_like=false
loss="cossim"
temperature=0.07
css_beta=0.5
# ============ Training ============= #
batch_size=3
max_iter=20000
# =================================== #
# Run training script
linear_text=""
if $linear ; then
linear_text="--linear"
fi
styleclip=""
if $styleclip_like ; then
styleclip="--styleclip"
fi
python train.py --gan=${gan} \
--truncation=0.7 \
--stylegan-space=${stylegan_space} \
--stylegan-layer=${stylegan_layer} \
--corpus=${corpus} \
--num-latent-support-dipoles=${num_latent_support_dipoles} \
--lss-beta=${lss_beta} \
--loss=${loss} \
--temperature=${temperature} \
--css-beta=${css_beta} \
${linear_text} \
${styleclip} \
--min-shift-magnitude=${min_shift_magnitude} \
--max-shift-magnitude=${max_shift_magnitude} \
--batch-size=${batch_size} \
--max-iter=${max_iter} \
--log-freq=10 \
--ckp-freq=100
|