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

     1  apiVersion: "kubeflow.org/v1"
     2  kind: TFJob
     3  metadata:
     4    name: tfjob-simple
     5    namespace: kubeflow
     6  spec:
     7    tfReplicaSpecs:
     8      Worker:
     9        replicas: 2
    10        restartPolicy: OnFailure
    11        template:
    12          spec:
    13            containers:
    14              - name: tensorflow
    15                image: kubeflow/tf-mnist-with-summaries:latest
    16                command:
    17                  - "python"
    18                  - "/var/tf_mnist/mnist_with_summaries.py"