diff --git a/utils/run_vm2.sh b/utils/run_vm2.sh index 047c9df40..625fb352d 100644 --- 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 \ No newline at end of file