Skip to content

Instantly share code, notes, and snippets.

@Chenhait

Chenhait/cpuman-perf-test-tf.yaml

Forked from balajismaniam/cpuman-perf-test-tf.yaml
Created May 21, 2022 08:43
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save Chenhait/9e7a5041bf542f47b38c9c179d8b9c56 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save Chenhait/9e7a5041bf542f47b38c9c179d8b9c56 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cpuman-perf-test-tf.yaml
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