Skip to content

Commit d5c00a4

Browse files
authored
Compliance run cmd update for yolo (#812)
1 parent 27d53d8 commit d5c00a4

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

script/app-mlperf-inference/customize.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,14 @@ def postprocess(i):
180180
out_baseline_accuracy_string = f""" --output-file {q}{os.path.join(output_dir, "accuracy", "baseline_accuracy.txt")}{q} """
181181
out_compliance_accuracy_string = f""" --output-file {q}{os.path.join(output_dir, "accuracy", "compliance_accuracy.txt")}{q} """
182182

183+
elif 'yolo' in model:
184+
accuracy_filename = "accuracy-coco.py"
185+
accuracy_filepath = os.path.join(
186+
env['MLC_MLPERF_INFERENCE_CLASSIFICATION_AND_DETECTION_PATH'], "tools", accuracy_filename)
187+
dataset_args = f""" --coco-dir {q}{env.get('MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH')}{q} """
188+
accuracy_log_file_option_name = " --mlperf-accuracy-file "
189+
datatype_option = ""
190+
183191
elif 'stable-diffusion-xl' in model:
184192
pass # No compliance check for now
185193
elif 'gpt' in model:

0 commit comments

Comments
 (0)