diff --git a/scripts/v1beta1/push.sh b/scripts/v1beta1/push.sh index 22bc8430b05..1d098815933 100755 --- a/scripts/v1beta1/push.sh +++ b/scripts/v1beta1/push.sh @@ -107,8 +107,11 @@ docker push "${REGISTRY}/enas-cnn-cifar10-gpu:${TAG}" echo -e "\nPushing Keras CIFAR-10 CNN training container example for ENAS with CPU support...\n" docker push "${REGISTRY}/enas-cnn-cifar10-cpu:${TAG}" -echo -e "\nPushing PyTorch CIFAR-10 CNN training container example for DARTS...\n" -docker push "${REGISTRY}/darts-cnn-cifar10:${TAG}" +echo -e "\nPushing PyTorch CIFAR-10 CNN training container example for DARTS with CPU support...\n" +docker push "${REGISTRY}/darts-cnn-cifar10-cpu:${TAG}" + +echo -e "\nPushing PyTorch CIFAR-10 CNN training container example for DARTS with GPU support...\n" +docker push "${REGISTRY}/darts-cnn-cifar10-gpu:${TAG}" echo -e "\nPushing dynamic learning rate training container example for PBT...\n" docker push "${REGISTRY}/simple-pbt:${TAG}"