diff --git a/perfmetrics/scripts/ml_tests/pytorch/run_model.sh b/perfmetrics/scripts/ml_tests/pytorch/run_model.sh index d230b9430d..556e7d4d41 100755 --- a/perfmetrics/scripts/ml_tests/pytorch/run_model.sh +++ b/perfmetrics/scripts/ml_tests/pytorch/run_model.sh @@ -16,9 +16,6 @@ PYTORCH_VERSION=$1 BUCKET_TYPE=$2 -echo "Pytorch version in run model: " $PYTORCH_VERSION -echo "Bucket typein run model: " $BUCKET_TYPE - NUM_EPOCHS=80 TEST_BUCKET="gcsfuse-ml-data" diff --git a/perfmetrics/scripts/ml_tests/pytorch/v2/dino/setup_host_and_run_container.sh b/perfmetrics/scripts/ml_tests/pytorch/v2/dino/setup_host_and_run_container.sh index 69bd95502f..6cfb5a7297 100755 --- a/perfmetrics/scripts/ml_tests/pytorch/v2/dino/setup_host_and_run_container.sh +++ b/perfmetrics/scripts/ml_tests/pytorch/v2/dino/setup_host_and_run_container.sh @@ -25,7 +25,5 @@ source ml_tests/setup_host.sh $DRIVER_VERSION PYTORCH_VERSION="v2" BUCKET_TYPE=$1 -echo "Pytorch version: " $PYTORCH_VERSION -echo "Bucket type: " $BUCKET_TYPE source ml_tests/pytorch/run_container.sh $PYTORCH_VERSION $BUCKET_TYPE