Opening PyTorch VAE… Continue