We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e446423 commit 6eed2e1Copy full SHA for 6eed2e1
biapy/engine/denoising.py
@@ -233,7 +233,7 @@ def metric_calculation(
233
234
with torch.no_grad():
235
for i, metric in enumerate(list_to_use):
236
- val = metric(_output, _targets[:, _output.shape[1]:].contiguous())
+ val = metric(_output.contiguous(), _targets[:, _output.shape[1]:].contiguous())
237
val = val.item() if not torch.isnan(val) else 0
238
out_metrics[list_names_to_use[i]] = val
239
0 commit comments