Skip to content

Instantly share code, notes, and snippets.

@snakers4

snakers4/train.sh

Created July 7, 2018 11:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save snakers4/dbef008b89ee56e21936f4897c286ddf to your computer and use it in GitHub Desktop.
Save snakers4/dbef008b89ee56e21936f4897c286ddf to your computer and use it in GitHub Desktop.
Download ZIP
VAE explanation bits
Raw
train.sh
python3 train.py \
--epochs 30 --batch-size 512 --seed 42 \
--model_type fc_conv --dataset_type fmnist --latent_space_size 10 \
--do_augs False \
--lr 1e-3 --m1 40 --m2 50 \
--optimizer adam \
--do_running_mean False --img_loss_weight 1.0 --kl_loss_weight 1.0 \
--image_loss_type bce --ssim_window_size 5 \
--print-freq 10 \
--lognumber fmnist_fc_conv_l10_rebalance_no_norm \
--tensorboard True --tensorboard_images True \
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment