Torch.nn.MSELoss

lastmod 2024-05-08created 2024-05-082 edits

A loss function that computes the mean squared error.

Usage

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

Example

import torch.nn as nn

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

See also

• pytorch

Favorite site

• MSELoss — PyTorch 2.3 documentation
• MSE Loss
• (PyTorch) Summary of commonly used Loss Functions (Cross-Entropy, MSE) - ZZEN’s Blog