Skip to content

Instantly share code, notes, and snippets.

@ynott

ynott/cpuman-perf-test-tf.yaml

Forked from balajismaniam/cpuman-perf-test-tf.yaml
Created September 29, 2018 12:49
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ynott/6e3cb138b245e827a6f9bae6f34fbe53 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ynott/6e3cb138b245e827a6f9bae6f34fbe53 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @balajismaniam balajismaniam created this gist Jul 17, 2018.
    37 changes: 37 additions & 0 deletions cpuman-perf-test-tf.yaml
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,37 @@
    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"