Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
apiVersion: v1
kind: Pod
metadata:
name: tf-wide-deep-census-pod
spec:
containers:
- image: <insert-image>
command: ["/bin/sh"]
args: ["-c", "/usr/bin/time python ./wide_deep.py --model_type=wide_deep --train_epochs=500 --epochs_between_eval=500 --batch_size=32561"]
name: tf-wide-deep-census
resources:
requests:
cpu: <insert-cpu-request>
memory: 32Gi
limits:
cpu: <insert-cpu-limit>
memory: 32Gi
restartPolicy: "Never"
---
apiVersion: v1
kind: Pod
metadata:
name: tf-resnet-cifar-pod
spec:
containers:
- image: <insert-image>
command: ["/bin/sh"]
args: ["-c", "/usr/bin/time python ./cifar10_main.py --resnet_size=50 --train_epochs=3 --batch_size=1000 --epochs_between_evals=3 --num_parallel_calls=24"]
name: tf-resnet-cifar
resources:
requests:
cpu: <insert-cpu-request>
memory: 128Gi
limits:
cpu: <insert-cpu-limit>
memory: 128Gi
restartPolicy: "Never"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.