add flags for saving results files

mlcommons · Jul 2, 2024 · 3212115 · 3212115
1 parent ea586fe
commit 3212115
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/scoring/score_submissions.py b/scoring/score_submissions.py
@@ -46,6 +46,16 @@
     'self_tuning_ruleset',
     False,
     'Whether to score on self-tuning ruleset or externally tuned ruleset')
+flags.DEFINE_string(
+    'save_results_to_filename',
+    None,
+    'Filename to save the processed results that are fed into the performance profile functions'
+)
+flags.DEFINE_boolean(
+    'load_results_from_filename',
+    None,
+    'Filename to load processed results from that are fed into performance profile functions'
+)
 FLAGS = flags.FLAGS