Skip to content

Commit 131f449

Browse files
authored
Update submission_checker.py | Fix open model unit in Results
1 parent c0b60cd commit 131f449

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/submission/submission_checker.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1977,7 +1977,7 @@ def log_result(
19771977
if config.version == "v4.0":
19781978
unit = unit_dict[scenario_fixed]
19791979
else:
1980-
unit = special_unit_dict.get(model_name, unit_dict)[scenario_fixed]
1980+
unit = special_unit_dict.get(mlperf_model, unit_dict)[scenario_fixed]
19811981
power_unit = power_unit_dict[scenario_fixed]
19821982

19831983
if (power_metric <= 0) or (

0 commit comments

Comments
 (0)