Skip to content

Commit b54ebaf

Browse files
authored
Update full dataset tag (#2104)
1 parent 7cfebee commit b54ebaf

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

main.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -755,9 +755,14 @@ def mlperf_inference_run_command(
755755
if "dlrm-v2" in model and implementation == "nvidia":
756756
cmd_suffix += f" \\\n{pre_space} --criteo_day23_raw_data_path=<PATH_TO_CRITEO_DAY23_RAW_DATA>"
757757

758+
if "short" in extra_variation_tags:
759+
full_ds_needed_tag = ""
760+
else:
761+
full_ds_needed_tag = "_full,"
762+
758763
run_cmd = f"""\n
759764
{f_pre_space}```bash
760-
{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} \\
761766
{pre_space} --model={model} \\
762767
{pre_space} --implementation={implementation} \\
763768
{pre_space} --framework={framework} \\

0 commit comments

Comments
 (0)