An error in ssim?

[YuningZhang]

There might be an error in train.py:
"TypeError: ssim() got an unexpected keyword argument 'reduction', (line 98, in train.py), ssim = 1 - 2 * kornia.losses.ssim(encoded_images.detach(), images, window_size=5, reduction="mean"). "
It seems that "ssim" should be "ssim_loss". Is it right?

[jzyustc]

Hi, the TypeError is caused by the different kornia version. It will be solved if you use kornia 0.3.0.

[YuningZhang]

Thank you very much!