set pytorch as default framework

mlcommons · Nov 13, 2024 · cf30e25 · cf30e25
1 parent aaa7b5c
commit cf30e25
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/script/generate-mlperf-inference-submission/customize.py b/script/generate-mlperf-inference-submission/customize.py
@@ -88,6 +88,11 @@ def generate_submission(i):
     results = [f for f in os.listdir(results_dir) if not os.path.isfile(os.path.join(results_dir, f))]
 
     system_meta_default = state['CM_SUT_META']
+
+    # set pytorch as the default framework
+    if system_meta_default['framework'] == '':
+        system_meta_default['framework'] = "pytorch"
+
     system_meta = {}
     if 'CM_MLPERF_SUBMISSION_SYSTEM_TYPE' in env:
         system_meta['system_type'] = env['CM_MLPERF_SUBMISSION_SYSTEM_TYPE']