Fix preprocess_submission script to copy the code path while inferrin…

…g low accuracy results (mlcommons#1627)
ctuning · Feb 20, 2024 · de31ee2 · de31ee2
1 parent 486a629
commit de31ee2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tools/submission/preprocess_submission.py b/tools/submission/preprocess_submission.py
@@ -247,6 +247,13 @@ def infer_scenario_results(filter_submitter, noinfer_low_accuracy_results, confi
 
                                     shutil.copytree(high_accuracy_model_path, \
                                             low_accuracy_model_path)
+                                high_accuracy_model_code_path = os.path.join(log_path, "..", \
+                                        "code", model)
+                                low_accuracy_model_code_path = os.path.join(log_path, "..", \
+                                        "code", low_accuracy_model)
+                                if not os.path.exists(low_accuracy_model_code_path):
+                                    shutil.copytree(high_accuracy_model_code_path, \
+                                            low_accuracy_model_code_path)