fixing error

mlcommons · Dec 19, 2024 · 1e9447c · 1e9447c
1 parent 8b9a797
commit 1e9447c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/GANDLF/cli/generate_metrics.py b/GANDLF/cli/generate_metrics.py
@@ -375,7 +375,8 @@ def __percentile_clip(
             if compute_ncc:
                 calculated_ncc_metrics = ncc_metrics(output_infill, gt_image_infill)
                 for key, value in calculated_ncc_metrics.items():
-                    overall_stats_dict[current_subject_id][key] = value.item()
+                    # we don't need the ".item()" here, since the values are already scalars
+                    overall_stats_dict[current_subject_id][key] = value
 
             # only voxels that are to be inferred (-> flat array)
             # these are required for mse, psnr, etc.