[Click on image for larger view.] Figure 2: Variational Autoencoder Architecture for the UCI Digits ... vae = VAE().to(device) # create model bat_size = 10 max_epochs = 10 log_interval = 2 lrn_rate = ...