LDM复现
1 环境下载
1.1 pytorch-lightning
pytorch-lightning==1.4.2
pytorch-lightning会自动下载高版本torch,所以应该先下这个,然后删掉高版本torch,再从官网下载对应的GPU版本的torch。(其中cudatoolkit pytorch torchvision要配套 ,查看是否配套的网址
1.2 taming安装位置
将taming-transform下载到E:\anaconda3\envs\wx004\Lib\site-packages 安装到conda wx004环境下的site-packages中,即可直接引用taming包。
其中self.quantize = VectorQuantizer(n_embed, embed_dim, beta=0.25, # remap=remap, # sane_index_shape=sane_index_shape)
在taming包中没有这两个参数(remap,sane_index_shape)检测到版本没有问题,所以删除参数就可以了。
1.3 训练autoencoder
出现 name 'trainer' is not defined
是因为科学上网导致!!!,导致网络不能下载\taming\modules\autoencoder\lpips\vgg.pth vgg.pth文件。解决方案是把梯子关了就可以。