vae_train.py 5.7 KB