Update full dataset tag (#2104)

mlcommons · Feb 11, 2025 · b54ebaf · b54ebaf
1 parent 7cfebee
commit b54ebaf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -755,9 +755,14 @@ def mlperf_inference_run_command(
             if "dlrm-v2" in model and implementation == "nvidia":
                 cmd_suffix += f" \\\n{pre_space} --criteo_day23_raw_data_path=<PATH_TO_CRITEO_DAY23_RAW_DATA>"
 
+            if "short" in extra_variation_tags:
+                full_ds_needed_tag = ""
+            else:
+                full_ds_needed_tag = "_full,"
+
             run_cmd = f"""\n
 {f_pre_space}```bash
-{f_pre_space}mlcr run-mlperf,inference,_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
+{f_pre_space}mlcr run-mlperf,inference,{full_ds_needed_tag}_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
 {pre_space} --model={model} \\
 {pre_space} --implementation={implementation} \\
 {pre_space} --framework={framework} \\