Merge branch 'master' into 689-no-need-for-the-history-file

mlcommons · Jul 7, 2023 · a6f96be · a6f96be
2 parents 5789aad + fdb73f8
commit a6f96be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/cli/generate_metrics.py b/GANDLF/cli/generate_metrics.py
@@ -225,7 +225,7 @@ def __fix_2d_tensor(input_tensor):
 
  overall_stats_dict[current_subject_id][
  "ssim"
- ] = structural_similarity_index(gt_image_infill, output_infill).item()
+ ] = structural_similarity_index(gt_image_infill, output_infill, mask).item()
 
  # ncc metrics
  overall_stats_dict[current_subject_id]["ncc_mean"] = ncc_mean(