remove local dependency

GoogleCloudPlatform · Aug 26, 2024 · 26c79ad · 26c79ad
1 parent 4a93062
commit 26c79ad
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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