Skip to content

Instantly share code, notes, and snippets.

@pingsutw
Created April 21, 2020 03:47
Show Gist options
  • Save pingsutw/55a6abbdbd0889315cee454fbe09766b to your computer and use it in GitHub Desktop.
Save pingsutw/55a6abbdbd0889315cee454fbe09766b to your computer and use it in GitHub Desktop.
kubeflow - TensorFlow with init container
apiVersion: "kubeflow.org/v1"
kind: "TFJob"
metadata:
name: "mnist"
namespace: submarine
spec:
cleanPodPolicy: None
initContainers:
- name: init-mydb
image: busybox:1.28
command: ['ls']
tfReplicaSpecs:
Worker:
replicas: 1
restartPolicy: Never
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"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment