modify run file

mlcommons · Apr 11, 2024 · aec4ce0 · aec4ce0
1 parent 5d97c7a
commit aec4ce0
Showing 1 changed file with 17 additions and 17 deletions.
diff --git a/utils/run_vm2.sh b/utils/run_vm2.sh
@@ -1,22 +1,17 @@
-python run_workloads.py \
---framework pytorch \
---docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
---workload_config_path submission_configs_small/AlgoPerf_Team_10_external_tuning_schedule_free_adamw.json \
+# python run_workloads.py \
+# --framework pytorch \
+# --docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
+# --workload_config_path submission_configs_small/AlgoPerf_Team_10_external_tuning_schedule_free_adamw.json \
 
-python run_workloads.py \
---framework pytorch \
---docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
---workload_config_path submission_configs_small/AlgoPerf_Team_10_external_tuning_schedule_free_prodigy.json
+# python run_workloads.py \
+# --framework pytorch \
+# --docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
+# --workload_config_path submission_configs_small/AlgoPerf_Team_10_external_tuning_schedule_free_prodigy.json
 
-python run_workloads.py \
---framework pytorch \
---docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
---workload_config_path submission_configs_small/AlgoPerf_Team_10_self_tuning_schedule_free_adamw.json
-
-python run_workloads.py \
---framework pytorch \
---docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
---workload_config_path submission_configs_small/AlgoPerf_Team_17_self_tuning_lwrkap.json
+# python run_workloads.py \
+# --framework pytorch \
+# --docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
+# --workload_config_path submission_configs_small/AlgoPerf_Team_10_self_tuning_schedule_free_adamw.json
 
 python run_workloads.py \
 --framework pytorch \
@@ -32,3 +27,8 @@ python run_workloads.py \
 --framework jax \
 --docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_jax_dev \
 --workload_config_path submission_configs_small/AlgoPerf_Team_16_external_tuning_caspr_adaptive.json
+
+python run_workloads.py \
+--framework pytorch \
+--docker_image_url us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_pytorch_dev \
+--workload_config_path submission_configs_small/AlgoPerf_Team_17_self_tuning_lwrkap.json