diff --git a/losses/RMSLE.py b/losses/RMSLE.py new file mode 100644 index 0000000000000000000000000000000000000000..cf5cff69c2f0421cdce0b80bf171d13e6b57489e --- /dev/null +++ b/losses/RMSLE.py @@ -0,0 +1,9 @@ +import torch.nn as nn + +class RMSLELoss(nn.Module): + def __init__(self): + super().__init__() + self.mse = nn.MSELoss() + + def forward(self, pred, actual): + return torch.sqrt(self.mse(torch.log(pred + 1), torch.log(actual + 1))) \ No newline at end of file