PyTorch VAE 在pytorch中实现的变体自动编码器(VAE)的集合,着重于可重复性。 该项目的目的是为其中许多很酷的VAE模型提供一个快速简单的工作示例。 所有模型都在上进行了训练,以一致性和比较性。 所有模型的架构在相同的层上都保持尽可能相似,除非原始论文需要根本不同的架构(例如,VQ VAE使用残差层,不使用批处理规范,这与其他模型不同)。 这是每个模型的。 要求 Python> = 3.5 PyTorch> = 1.3 Pytorch Lightning> = 0.6.0( ) 启用CUDA的计算设备 安装 $ git clone https://github.co