PyTorch-VAE:多种变分自编码器实现,支持PyTorch Lightning,CelebA数据集训练