复现Beta-VAE

复现Beta-VAE

文章目录

• 前言
• 问题一、visdom的配置
• 问题二、IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item＜T＞()` in
• 问题三、TypeError: save_image() missing 1 required positional argument: 'fp'
• 问题四、/bin/sh: 1: convert: not found
• 问题五、Python3.8安装PIL包

---

前言

这是复现Beta-vae的踩坑之旅，代码链接

---

问题一、visdom的配置

该代码需要使用visdom，关于如何配置visdom可以看这篇文章visdom远程连接

问题二、IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item＜T＞() in

遇到这个问题解决方法

---

问题三、TypeError: save_image() missing 1 required positional argument: ‘fp’

在复现beta-vae时，我遇到了一些问题，每当跌倒到10000次的时候，需要保存生成的图片，这时solver.py的第417行save_image(tensor=gifs[i][j].cpu(),都会报一个错误TypeError: save_image() missing 1 required positional argument: 'fp

解决方法：
我找到save_image()的源码，然后直接将源码放到class中，直接使用save_image()方法，而不是通过 from torchvision.utils import save_image 调用包来使用

问题四、/bin/sh: 1: convert: not found

遇到这个问题解决方法

sudo apt-get install imagemagick

问题五、Python3.8安装PIL包

目前PIL只支持Python2.x版本
Python3.x版本需要使用pillow

pip install pillow