I think it's not necessary to judge if X.type() == Y.type()

[Do1e]

pytorch-msssim/pytorch_msssim/ssim.py

Line 135 in d23a69e

if not X.type() == Y.type():

pytorch-msssim/pytorch_msssim/ssim.py

Line 183 in d23a69e

if not X.type() == Y.type():

I think it's not necessary to judge if X.type() == Y.type(), or mixed precision training will not be supported.