持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第15天，点击查看活动详情

这的复现不是说咱们从头到尾自己写啊，我就说一下怎么用人家的仓库无脑跑起来的。

要求

• linux
• 有GPU、可以用CUDA（这些都是必须不用多说）
• python 3.8
• PyTorch >= 1.3
• Pytorch Lightning >= 0.6.0 (GitHub Repo)

服务器

虚拟环境

先用conda创建好对应的虚拟环境

conda create –n vae python=3.8
conda activate vae

代码准备

1. 先把代码下载到本地。

2. 安装所需要的包

   cd PyTorch-VAE
   pip install -r requirements.txt
   

   先进入到刚才clone下来的仓库里再安装嗷。记得打开虚拟环境了再安装！

   

3. 配置文件

   

   看一下有一个configs文件夹，底下就是各种VAE模型极其变体。

   

   我们要跑的是VAE的基础模型， 也就是要对vae.yaml进行修改。要改的没什么，我这里用的是32G的V100所以batch size这里设置的比较大，大家按需修改一下。

   

4. 运行代码

   python run.py -c configs/<config-file-name.yaml>
   

   

实验结果

实验结果可以直接用tensorboard打开。

训练集结果：

验证集结果：

生成效果：

左边采样，右边重建结果。

AntixK/PyTorch-VAE at 8700d245a9735640dda458db4cf40708caf2e77f (github.com)