ssim functional metric not working with 16 bit precision
🐛 Bug
ssim functional metric doesn't work with native 16 bit precision
Please reproduce using the BoringModel
https://colab.research.google.com/drive/1Wzxjq-oQavT-kg9go9Ti-AcCEhuir7H9?usp=sharing
if you remove 16 bit precision from trainer it works but with 16 bit precision it gives the following error:
TypeError: Expected `preds` and `target` to have the same data type. Got pred: torch.float32 and target: torch.float16.