reconstruct(vqvae, img, device, cfg['dataset_type']) 训练压缩图像生成模型 PixelCNN 有了一个VQVAE后,我们要用另一个模型对VQVAE的离散空间采样,也就是训练一个能生成压缩图片的模型。我们可以按照VQVAE论文的方法,使用PixelCNN来生成压缩图片。 PixelCNN 的原理及实现方法就不在这里过多介绍了。