File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
script/app-mlperf-inference Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments