Skip to content

Commit 318cb13

Browse files
authored
Tuples (#1234)
* Support tuples for accelerator count * Support ssd-large in open
1 parent 1749731 commit 318cb13

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tools/submission/generate-final-report.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@ def main():
4949

5050
# cleanup counts
5151
df['Accelerator'] = df['Accelerator'].apply(lambda x: x if x != '-' else '')
52-
df['a#'] = df['a#'].apply(lambda x: int(x) if x != '' else 0)
53-
df['a#'] = df['a#'].apply(lambda x: x if x > 0 else '')
52+
df['a#'] = df['a#'].apply(lambda x: int(x) if x.isnumeric() else x)
53+
df['a#'] = df['a#'].apply(lambda x: x if x != 0 else '')
5454
df['p#'] = df.apply(lambda x: int(x['host_processors_per_node']), axis=1)
5555

5656
# details url

tools/submission/submission-checker.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -868,7 +868,7 @@
868868
"model_mapping": {
869869
# map model names to the official mlperf model class
870870
"ssd-mobilenet": "ssd-small",
871-
"ssd-resnet34": "ssd-large",
871+
"ssd-resnet34": "retinanet",
872872
"mobilenet": "resnet",
873873
"resnet50": "resnet",
874874
"ssd_resnet101_v1_fpn_640x640": "ssd-small",
@@ -1820,7 +1820,7 @@ def log_result(submitter,
18201820
system_json.get("host_processors_per_node"),
18211821
system_json.get("host_processor_core_count"),
18221822
'"' + system_json.get("accelerator_model_name") + '"',
1823-
system_json.get("accelerators_per_node"),
1823+
'"' + str(system_json.get("accelerators_per_node")) + '"',
18241824
name.replace("\\", "/"),
18251825
'"' + system_json.get("framework", "") + '"',
18261826
'"' + system_json.get("operating_system", "") + '"',
@@ -1837,7 +1837,7 @@ def log_result(submitter,
18371837
system_json.get("host_processors_per_node"),
18381838
system_json.get("host_processor_core_count"),
18391839
'"' + system_json.get("accelerator_model_name") + '"',
1840-
system_json.get("accelerators_per_node"),
1840+
'"' + str(system_json.get("accelerators_per_node")) + '"',
18411841
name.replace("\\", "/"),
18421842
'"' + system_json.get("framework", "") + '"',
18431843
'"' + system_json.get("operating_system", "") + '"',

0 commit comments

Comments
 (0)