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