How to use RMSE loss function in PyTorch

rmse_loss.py

# Thanks https://discuss.pytorch.org/t/rmse-loss-function/16540

class RMSELoss(torch.nn.Module):
    def __init__(self):
        super(RMSELoss,self).__init__()

    def forward(self,x,y):
        criterion = nn.MSELoss()
        loss = torch.sqrt(criterion(x, y))
        return loss

You need to add an epsilone in case of 0, as in backpropagation it will result in nans!
for example sth like this:

eps = 1e-6
loss = torch.sqrt(criterion(x, y) + eps)

That was incredibly useful thanks !