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"