github.com/kubeflow/training-operator@v1.7.0/examples/tensorflow/dist-mnist/tf_job_mnist.yaml (about)

     1  apiVersion: "kubeflow.org/v1"
     2  kind: "TFJob"
     3  metadata:
     4    name: "dist-mnist-for-e2e-test"
     5  spec:
     6    tfReplicaSpecs:
     7      PS:
     8        replicas: 2
     9        restartPolicy: Never
    10        template:
    11          spec:
    12            containers:
    13              - name: tensorflow
    14                image: kubeflow/tf-dist-mnist-test:latest
    15      Worker:
    16        replicas: 4
    17        restartPolicy: Never
    18        template:
    19          spec:
    20            containers:
    21              - name: tensorflow
    22                image: kubeflow/tf-dist-mnist-test:latest