ChanghoWoo
21 lines
573 B
YAML
21 lines
573 B
YAML
apiVersion: "kubeflow.org/v1"
|
|
kind: TFJob
|
|
metadata:
|
|
name: tfjob-simple
|
|
namespace: kubeflow
|
|
spec:
|
|
tfReplicaSpecs:
|
|
Worker:
|
|
replicas: 2
|
|
restartPolicy: OnFailure
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: tensorflow
|
|
image: gcr.io/kubeflow-ci/tf-mnist-with-summaries:1.0
|
|
command:
|
|
- "python"
|
|
- "/var/tf_mnist/mnist_with_summaries.py"
|
|
- "--log_dir=/train/logs"
|
|
- "--learning_rate=0.01"
|
|
- "--batch_size=150" |