Update pytorch example with latest image (#329)

* Update pytorch example with latest image * Update pytorch example docker image
kubeflow · Jan 18, 2019 · 15bbcae · 15bbcae
1 parent a24c428
commit 15bbcae
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/pytorchjob-example.yaml b/examples/pytorchjob-example.yaml
@@ -43,11 +43,11 @@ spec:
  spec:
  containers:
  - name: pytorch
- image: gcr.io/kubeflow-ci/pytorch-mnist-with-summary:1.0
+ image: gcr.io/kubeflow-ci/pytorch-dist-mnist-test:1.0
  imagePullPolicy: Always
  command:
  - "python"
- - "/opt/pytorch_dist_mnist/dist_mnist_with_summary.py"
+ - "/var/mnist.py"
  {{- with .HyperParameters}}
  {{- range .}}
  - "{{.Name}}={{.Value}}"
@@ -60,11 +60,11 @@ spec:
  spec:
  containers:
  - name: pytorch
- image: gcr.io/kubeflow-ci/pytorch-mnist-with-summary:1.0
+ image: gcr.io/kubeflow-ci/pytorch-dist-mnist-test:1.0
  imagePullPolicy: Always
  command:
  - "python"
- - "/opt/pytorch_dist_mnist/dist_mnist_with_summary.py"
+ - "/var/mnist.py"
  {{- with .HyperParameters}}
  {{- range .}}
  - "{{.Name}}={{.Value}}"