From aa6fca5536196fab6c8d107e79cdf92a01ac9aa1 Mon Sep 17 00:00:00 2001 From: Tulsi Shah Date: Mon, 26 Aug 2024 16:37:19 +0000 Subject: [PATCH] small fix --- perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile b/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile index b7cef4f377..b92732b215 100644 --- a/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile +++ b/perfmetrics/scripts/ml_tests/pytorch/v2/dino/Dockerfile @@ -41,7 +41,8 @@ RUN mkdir -p "gcsfuse_data" ARG PYTORCH_VERSION ARG BUCKET_TYPE +ENV PYTORCH_VERSION=${PYTORCH_VERSION} +ENV BUCKET_TYPE=${BUCKET_TYPE} -RUN echo ${PYTORCH_VERSION} RUN echo ${BUCKET_TYPE} ENTRYPOINT ["/bin/bash", "-c", "./run_model.sh ${PYTORCH_VERSION} ${BUCKET_TYPE}"]