From 470bc7b26a90ba6389a1f12e7bb99e2cbd72bd3c Mon Sep 17 00:00:00 2001 From: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com> Date: Tue, 11 Feb 2025 21:08:35 +0530 Subject: [PATCH] Update full dataset tag --- main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 776f42b28..b442c39f7 100755 --- 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=" + 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} \\