File size: 368 Bytes
16906c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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/" |