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

     1  apiVersion: "kubeflow.org/v1"
     2  kind: "TFJob"
     3  metadata:
     4    name: "mnist"
     5    namespace: kubeflow 
     6  spec:
     7    runPolicy:
     8      cleanPodPolicy: None
     9    tfReplicaSpecs:
    10      Worker:
    11        replicas: 1 
    12        restartPolicy: Never
    13        template:
    14          spec:
    15            containers:
    16              - name: tensorflow
    17                image: kubeflow/tf-mnist-with-summaries:latest
    18                command:
    19                  - "python"
    20                  - "/var/tf_mnist/mnist_with_summaries.py"
    21                  - "--log_dir=/train/logs"
    22                  - "--learning_rate=0.01"
    23                  - "--batch_size=150"
    24                volumeMounts:
    25                  - mountPath: "/train"
    26                    name: "training"
    27            volumes:
    28              - name: "training"
    29                persistentVolumeClaim:
    30                  claimName: "tfevent-volume"