9.2.6 训练VAE模型