Skip to content

Commit d35914a

Browse files
authored
Merge branch 'master' into bugfix/pearson_single_update
2 parents b389158 + 42c748b commit d35914a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

CHANGELOG.md

+3
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
2929
- Fixed bug in `PearsonCorrCoef` is updated on single samples at a time ([#2019](https://github.com/Lightning-AI/torchmetrics/pull/2019)
3030

3131

32+
- Fixed support for pixelwise MSE ([#2017](https://github.com/Lightning-AI/torchmetrics/pull/2017)
33+
34+
3235
## [1.1.0] - 2023-08-22
3336

3437
### Added

src/torchmetrics/functional/regression/mse.py

-2
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
import torch
1717
from torch import Tensor
1818

19-
from torchmetrics.functional.regression.utils import _check_data_shape_to_num_outputs
2019
from torchmetrics.utilities.checks import _check_same_shape
2120

2221

@@ -32,7 +31,6 @@ def _mean_squared_error_update(preds: Tensor, target: Tensor, num_outputs: int)
3231
3332
"""
3433
_check_same_shape(preds, target)
35-
_check_data_shape_to_num_outputs(preds, target, num_outputs, allow_1d_reshape=True)
3634
if num_outputs == 1:
3735
preds = preds.view(-1)
3836
target = target.view(-1)

0 commit comments

Comments
 (0)