#!/bin/bash


# === Experiment configuration ===
gan_type="SNGAN_AnimeFaces"
learn_alphas=false
learn_gammas=true
num_support_sets=64
num_support_dipoles=128
min_shift_magnitude=0.25
max_shift_magnitude=0.35
reconstructor_type="LeNet"
batch_size=128
max_iter=120000
tensorboard=true
# ================================


# Run training script
learn_a=""
if $learn_alphas ; then
  learn_a="--learn-alphas"
fi

learn_g=""
if $learn_gammas ; then
  learn_g="--learn-gammas"
fi

tb=""
if $tensorboard ; then
  tb="--tensorboard"
fi

python train.py $tb \
                --gan-type=${gan_type} \
                --reconstructor-type=${reconstructor_type} \
                $learn_a \
                $learn_g \
                --num-support-sets=${num_support_sets} \
                --num-support-dipoles=${num_support_dipoles} \
                --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
