training_params:
  max_epochs: 30
  auto_lr_find: false
  gpus: 1
model_params:
  model_type: conv-vae # vae or conv-vae
  lr: 0.005
  batch_size: 1
  hidden_size: 4096
  latent_size: 128
  alpha: 1024
  dataset: "fashion-mnist"
  save_images: true
  save_path: "log_images/"
  channels: 1
  height: 64
  width: 64
logger_params:
  name: "conv-vae"
  save_dir: "logs/"