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