diff --git a/src/ptbench/engine/saliency/evaluator.py b/src/ptbench/engine/saliency/evaluator.py index ddbe8013c3e5a0e31ad1ab4ba21e082b89b8fbb7..f8a9c04eb46cfc4b17003b2ccbe1873d6e987a4b 100644 --- a/src/ptbench/engine/saliency/evaluator.py +++ b/src/ptbench/engine/saliency/evaluator.py @@ -59,7 +59,7 @@ def _reconcile_metrics( continue aopc_combined = compl_info[5] - prop_energy = interp_info[4] + prop_energy = interp_info[2] road_weighted_prop_energy = max(0, aopc_combined) * prop_energy retval.append(