Add NAS RL training container to kubeflowkatib repository (#1008)

* Push nas training container to kubeflowkatib * Change README * Add line to readme * Fix link to mxnet-mnist example
kubeflow · Jan 13, 2020 · 74255c4 · 74255c4
1 parent 0df05e9
commit 74255c4
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 49 deletions.
diff --git a/examples/v1alpha3/README.md b/examples/v1alpha3/README.md
@@ -33,7 +33,7 @@ $ kubectl apply -f grid-example.yaml
 ```
 #### Bayesian Optimization Suggestion Demo
 ```
-$ kubectl apply -f bayseopt-example.yaml
+$ kubectl apply -f bayesianoptimization-example.yaml
 ```
 #### Hyperband Suggestion Demo
 ```
@@ -308,3 +308,41 @@ Acsess to `http://127.0.0.1:8000/katib`
 ## Clean
 Clean up with [destroy.sh](./MinikubeDemo/destroy.sh) script.
 It will stop port-forward process and delete minikube cluster.
+
+# List of current Katib training container images
+
+- Mxnet mnist example with collecting metrics time, [source](https://github.com/kubeflow/katib/blob/master/examples/v1alpha3/mxnet-mnist/mnist.py).
+
+```
+docker.io/kubeflowkatib/mxnet-mnist
+```
+
+- Pytorch mnist example with saving metrics to the file, [source](https://github.com/kubeflow/katib/blob/master/examples/v1alpha3/file-metrics-collector/mnist.py).
+
+```
+docker.io/kubeflowkatib/pytorch-mnist
+```
+
+- Keras cifar10 example for NAS RL with gpu support, [source](https://github.com/kubeflow/katib/blob/master/examples/v1alpha3/NAS-training-containers/RL-cifar10/Dockerfile.cpu).
+
+```
+docker.io/kubeflowkatib/nasrl-cifar10-gpu
+```
+
+- Keras cifar10 example for NAS RL with cpu support, [source](https://github.com/kubeflow/katib/blob/master/examples/v1alpha3/NAS-training-containers/RL-cifar10/Dockerfile.cpu).
+
+```
+docker.io/kubeflowkatib/nasrl-cifar10-cpu
+```
+
+- Pytorch operator mnist example, [source](https://github.com/kubeflow/pytorch-operator/blob/master/examples/mnist/mnist.py).
+
+```
+gcr.io/kubeflow-ci/pytorch-dist-mnist-test
+```
+
+- Tf operator mnist example, [source](https://github.com/kubeflow/tf-operator/blob/master/examples/v1/mnist_with_summaries/mnist_with_summaries.py).
+
+```
+gcr.io/kubeflow-ci/tf-mnist-with-summaries
+```
diff --git a/examples/v1alpha3/nasjob-example-RL.yaml b/examples/v1alpha3/nasjob-example-RL.yaml
@@ -60,7 +60,7 @@ spec:
               spec:
                 containers:
                 - name: {{.Trial}}
-                  image: gcr.io/kubeflow-images-public/katib/v1alpha3/training-container-nasrl-cifar10
+                  image: docker.io/kubeflowkatib/nasrl-cifar10-gpu
                   command:
                   - "python3.5"
                   - "-u"

diff --git a/test/scripts/v1alpha3/build-tc-nasrl-cifar10.sh b/test/scripts/v1alpha3/build-tc-nasrl-cifar10.sh
diff --git a/test/workflows/components/workflows-v1alpha3.libsonnet b/test/workflows/components/workflows-v1alpha3.libsonnet
@@ -251,10 +251,6 @@
                     name: "create-pr-symlink",
                     template: "create-pr-symlink",
                   },
-                  //{
-                  //  name: "build-tc-nasrl-cifar10",
-                  //  template: "build-tc-nasrl-cifar10",
-                  //},
                 ],
                 [
                   {
@@ -431,9 +427,6 @@
             $.parts(namespace, name, overrides).e2e(prow_env, bucket).buildTemplate("build-ui", testWorkerImage, [
               "test/scripts/v1alpha3/build-ui.sh",
             ]),  // build-ui
-            $.parts(namespace, name, overrides).e2e(prow_env, bucket).buildTemplate("build-tc-nasrl-cifar10", testWorkerImage, [
-              "test/scripts/v1alpha3/build-tc-nasrl-cifar10.sh",
-            ]),  // build-tc-nasrl-cifar10
           ],  // templates
         },
       },  // e2e