文章目录 前言解决方法: 前言 运行Beta-VAE代码时遇到这个问题,发现网上对这个问题的讨论还是很少的,这里我提供一个解决方法,不保证一定有用。 解决方法: 解决前:是通过导入from torchvision.utils import make_grid, save_image 来直接调用save_image方法,报如题所示的错误 解决后:找到save_image()的源码,然后直接将源码放到class中,直接使用save_image()方法。 修改后代码如图