chore: add pbt to repository build tools

Author: a9p

Makefile

@@ -110,6 +110,7 @@ prepare-pytest:
 	pip install -r cmd/suggestion/hyperband/v1beta1/requirements.txt
 	pip install -r cmd/suggestion/nas/enas/v1beta1/requirements.txt
 	pip install -r cmd/suggestion/nas/darts/v1beta1/requirements.txt
+	pip install -r cmd/suggestion/pbt/v1beta1/requirements.txt
 	pip install -r cmd/earlystopping/medianstop/v1beta1/requirements.txt
 	pip install -r cmd/metricscollector/v1beta1/tfevent-metricscollector/requirements.txt
 

prow_config.yaml

@@ -40,6 +40,7 @@ workflows:
       - cmd/suggestion/hyperopt/v1beta1/*
       - cmd/suggestion/nas/darts/v1beta1/*
       - cmd/suggestion/nas/enas/v1beta1/*
+      - cmd/suggestion/pbt/v1beta1/*
       - cmd/suggestion/skopt/v1beta1/*
       - cmd/ui/v1beta1/*
       - examples/v1beta1/*

scripts/v1beta1/build.sh

@@ -105,6 +105,9 @@ docker build --platform "linux/$ARCH" -t "${REGISTRY}/suggestion-enas:${TAG}" -f
 echo -e "\nBuilding DARTS suggestion...\n"
 docker build --platform "linux/$ARCH" -t "${REGISTRY}/suggestion-darts:${TAG}" -f ${CMD_PREFIX}/suggestion/nas/darts/${VERSION}/Dockerfile .
 
+echo -e "\nBuilding PBT suggestion...\n"
+docker build --platform "linux/$ARCH" -t "${REGISTRY}/suggestion-pbt:${TAG}" -f ${CMD_PREFIX}/suggestion/pbt/${VERSION}/Dockerfile .
+
 # Early stopping images
 echo -e "\nBuilding early stopping images...\n"
 
@@ -136,6 +139,9 @@ else
   echo -e "\nBuilding PyTorch CIFAR-10 CNN training container example for DARTS...\n"
   docker build --platform linux/amd64 -t "${REGISTRY}/darts-cnn-cifar10:${TAG}" -f examples/${VERSION}/trial-images/darts-cnn-cifar10/Dockerfile .
 
+  echo -e "\nBuilding dynamic learning rate training container example for PBT...\n"
+  docker build --platform linux/amd64 -t "${REGISTRY}/simple-pbt:${TAG}" -f examples/${VERSION}/trial-images/pbt/Dockerfile .
+
 fi
 
 echo -e "\nAll Katib images with ${TAG} tag have been built successfully!\n"

scripts/v1beta1/push.sh

@@ -80,6 +80,9 @@ docker push "${REGISTRY}/suggestion-enas:${TAG}"
 echo -e "\nPushing DARTS suggestion...\n"
 docker push "${REGISTRY}/suggestion-darts:${TAG}"
 
+echo -e "\nPushing PBT suggestion...\n"
+docker push "${REGISTRY}/suggestion-pbt:${TAG}"
+
 # Early stopping images
 echo -e "\nPushing early stopping images...\n"
 
@@ -107,4 +110,7 @@ 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 dynamic learning rate training container example for PBT...\n"
+docker push "${REGISTRY}/simple-pbt:${TAG}"
+
 echo -e "\nAll Katib images with ${TAG} tag have been pushed successfully!\n"