Torch.nn.MSELoss

mean squared error 를 구하는 loss function 입니다.

Usage

torch.nn.MSELoss(size_average=None, reduce=None, reduction='mean')

import torch.nn as nn

criterion = nn.MSELoss()
...
loss = criterion(input, target)