diff --git a/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py b/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py index 7ef577c8ec3b092b68e383edae07c6df2e1ac2d0..c0ef3bc66c3b25a3162630cc1e0fb4a1c4783310 100644 --- a/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py +++ b/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py @@ -315,7 +315,7 @@ class MSComparator(Comparator): for mapping_dict in self.api_mapping_dict: keys_to_compare = [ ('ms_args', 'pt_args'), - ('ms_output', 'pt_output'), + ('ms_outputs', 'pt_outputs'), ('ms_parameters', 'pt_parameters'), ('ms_parameters_grad', 'pt_parameters_grad'), ]