small fix

GoogleCloudPlatform · Aug 26, 2024 · aa6fca5 · aa6fca5
1 parent 26c79ad
commit aa6fca5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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}"]