Skip to content

Commit 9b32a74

Browse files
authored
Changes for final report generation - PointPainting (#2063)
* Update perf sample count default value * Updation for pointpainting * Add default value for model name argument
1 parent 115dd5b commit 9b32a74

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

automotive/3d-object-detection/main.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,10 @@ def get_args():
9292
help="enable finding peak performance pass",
9393
)
9494
parser.add_argument("--backend", help="Name of the backend")
95-
parser.add_argument("--model-name", help="Name of the model")
95+
parser.add_argument(
96+
"--model-name",
97+
help="Name of the model",
98+
default="pointpainting")
9699
parser.add_argument("--output", default="output", help="test results")
97100
parser.add_argument("--qps", type=int, help="target qps")
98101
parser.add_argument("--lidar-path", help="Path to model weights")
@@ -128,7 +131,7 @@ def get_args():
128131
parser.add_argument("--count", type=int, help="dataset items to use")
129132
parser.add_argument("--debug", action="store_true", help="debug")
130133
parser.add_argument(
131-
"--performance-sample-count", type=int, help="performance sample count", default=5000
134+
"--performance-sample-count", type=int, help="performance sample count", default=1024
132135
)
133136
parser.add_argument(
134137
"--max-latency", type=float, help="mlperf max latency in pct tile"

tools/submission/generate_final_report.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@ def main():
150150
"llama2-70b-interactive-99",
151151
"llama2-70b-interactive-99.9",
152152
"mixtral-8x7b",
153+
"pointpainting",
153154
],
154155
["SingleStream", "MultiStream", "Server", "Offline"],
155156
[
@@ -228,6 +229,7 @@ def main():
228229
"gptj-99": ["SingleStream", "Offline"],
229230
"gptj-99.9": ["SingleStream", "Offline"],
230231
"stable-diffusion-xl": ["SingleStream", "Offline"],
232+
"pointpainting": ["SingleStream"],
231233
},
232234
}
233235

tools/submission/truncate_accuracy_log.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,9 @@ def truncate_results_dir(filter_submitter, backup, scenarios_to_skip):
142142
continue
143143

144144
# process results
145-
required_dirs = ["results", "compliance"] if division in ["closed", "network"] else ["results"]
145+
required_dirs = [
146+
"results", "compliance"] if division in [
147+
"closed", "network"] else ["results"]
146148
for directory in required_dirs:
147149

148150
log_path = os.path.join(division, submitter, directory)

0 commit comments

Comments
 (0)