update kubeflow dip-catalog
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user