From 26c79ad7a34a7e49be9a363dba97127ac068f73d Mon Sep 17 00:00:00 2001 From: Tulsi Shah Date: Mon, 26 Aug 2024 15:39:22 +0000 Subject: [PATCH] remove local dependency --- perfmetrics/scripts/ml_tests/pytorch/run_model.sh | 4 ++++ perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile | 2 ++ .../ml_tests/pytorch/v2/dino/setup_host_and_run_container.sh | 3 +++ 3 files changed, 9 insertions(+) diff --git a/perfmetrics/scripts/ml_tests/pytorch/run_model.sh b/perfmetrics/scripts/ml_tests/pytorch/run_model.sh index 058367f1fa..d230b9430d 100755 --- a/perfmetrics/scripts/ml_tests/pytorch/run_model.sh +++ b/perfmetrics/scripts/ml_tests/pytorch/run_model.sh @@ -15,6 +15,10 @@ 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/Dockerfile b/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile index 38c1e905b9..b7cef4f377 100644 --- a/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile +++ b/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile @@ -42,4 +42,6 @@ RUN mkdir -p "gcsfuse_data" ARG PYTORCH_VERSION ARG BUCKET_TYPE +RUN echo ${PYTORCH_VERSION} +RUN echo ${BUCKET_TYPE} ENTRYPOINT ["/bin/bash", "-c", "./run_model.sh ${PYTORCH_VERSION} ${BUCKET_TYPE}"] 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 2891480b4a..69bd95502f 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,4 +25,7 @@ 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