small fix

GoogleCloudPlatform · Aug 27, 2024 · 6be9e62 · 6be9e62
1 parent e3e632e
commit 6be9e62
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/perfmetrics/scripts/ml_tests/pytorch/v1_12/dino/Dockerfile b/perfmetrics/scripts/ml_tests/pytorch/v1_12/dino/Dockerfile
@@ -31,5 +31,7 @@ RUN mkdir -p "gcsfuse_data"
 
 ARG PYTORCH_VERSION
 ARG BUCKET_TYPE
+ENV PYTORCH_VERSION=${PYTORCH_VERSION}
+ENV BUCKET_TYPE=${BUCKET_TYPE}
 
 ENTRYPOINT ["/bin/bash", "-c", "./run_model.sh ${PYTORCH_VERSION} ${BUCKET_TYPE}"]