I am currently trying to get another loss function to work, but I always get the same error:
Found dtype Double but expected Float
But how is that possible?
class MSLELoss(nn.Module):
    def __init__(self):
          super().__init__()
          self.mse = nn.MSELoss()
        
      def forward(self, pred, actual):
          return self.mse(pred,actual)

criterion = nn.MSELoss() works fine while
criterion = MSLELoss() does not…
What am I missing?
Thanks in advance!