We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7cfebee commit b54ebafCopy full SHA for b54ebaf
main.py
@@ -755,9 +755,14 @@ def mlperf_inference_run_command(
755
if "dlrm-v2" in model and implementation == "nvidia":
756
cmd_suffix += f" \\\n{pre_space} --criteo_day23_raw_data_path=<PATH_TO_CRITEO_DAY23_RAW_DATA>"
757
758
+ if "short" in extra_variation_tags:
759
+ full_ds_needed_tag = ""
760
+ else:
761
+ full_ds_needed_tag = "_full,"
762
+
763
run_cmd = f"""\n
764
{f_pre_space}```bash
-{f_pre_space}mlcr run-mlperf,inference,_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
765
+{f_pre_space}mlcr run-mlperf,inference,{full_ds_needed_tag}_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
766
{pre_space} --model={model} \\
767
{pre_space} --implementation={implementation} \\
768
{pre_space} --framework={framework} \\
0 commit comments