情况
在跑latent-diffusion时,训练编码器时,将原本的RGB输入改成单通道输入时报错,
报错显示在 LPIPSWithDiscriminator,回溯是“non-singleton dimension dismatch’ at line 154 'return (inp - self.shift) / self.scale”。
分享原因:
这个报错虽然在“return (inp - self.shift) / self.scale”,
但是解决方法不在此处。
解决方法
该解决办法存在的问题
参考:https://github.com/richzhang/PerceptualSimilarity/issues/127
https://github.com/richzhang/PerceptualSimilarity/issues/29
原始LPIPS这个计算loss的方式是对两个RGB图像,而不是对两个灰度图像,
从报错的这里也可以看出来。