diff --git a/examples/mnist/v1/pytorch_job_mnist_gloo.yaml b/examples/mnist/v1/pytorch_job_mnist_gloo.yaml index a1cef5721..2d039755f 100644 --- a/examples/mnist/v1/pytorch_job_mnist_gloo.yaml +++ b/examples/mnist/v1/pytorch_job_mnist_gloo.yaml @@ -14,7 +14,7 @@ spec: spec: containers: - name: pytorch - image: gcr.io//pytorch_dist_mnist:latest + image: gcr.io//pytorch-dist-mnist:latest args: ["--backend", "gloo"] # Comment out the below resources to use the CPU. resources: @@ -30,7 +30,7 @@ spec: spec: containers: - name: pytorch - image: gcr.io//pytorch_dist_mnist:latest + image: gcr.io//pytorch-dist-mnist:latest args: ["--backend", "gloo"] # Comment out the below resources to use the CPU. resources: diff --git a/examples/mnist/v1/pytorch_job_mnist_mpi.yaml b/examples/mnist/v1/pytorch_job_mnist_mpi.yaml index 5ee167b05..2b0ad5c4d 100644 --- a/examples/mnist/v1/pytorch_job_mnist_mpi.yaml +++ b/examples/mnist/v1/pytorch_job_mnist_mpi.yaml @@ -14,7 +14,7 @@ spec: spec: containers: - name: pytorch - image: gcr.io//pytorch_dist_mnist:mpi + image: gcr.io//pytorch-dist-mnist:mpi args: ["--backend", "mpi"] # Comment out the below resources to use the CPU. resources: @@ -30,7 +30,7 @@ spec: spec: containers: - name: pytorch - image: gcr.io//pytorch_dist_mnist:mpi + image: gcr.io//pytorch-dist-mnist:mpi args: ["--backend", "mpi"] # Comment out the below resources to use the CPU. resources: diff --git a/examples/mnist/v1/pytorch_job_mnist_nccl.yaml b/examples/mnist/v1/pytorch_job_mnist_nccl.yaml index bfa4889f4..65992127b 100644 --- a/examples/mnist/v1/pytorch_job_mnist_nccl.yaml +++ b/examples/mnist/v1/pytorch_job_mnist_nccl.yaml @@ -14,7 +14,7 @@ spec: spec: containers: - name: pytorch - image: gcr.io//pytorch_dist_mnist:latest + image: gcr.io//pytorch-dist-mnist:latest args: ["--backend", "nccl"] resources: limits: @@ -29,7 +29,7 @@ spec: spec: containers: - name: pytorch - image: gcr.io//pytorch_dist_mnist:latest + image: gcr.io//pytorch-dist-mnist:latest args: ["--backend", "nccl"] resources: limits: