beta_vae.zip