apiVersion: v1
kind: Pod
metadata:
  name: mnist-tensorflow-keras
  labels:
    app: mnist
    framework: tensorflow
spec:
  restartPolicy: OnFailure
  containers:
  - name: mnist-tensorflow-keras
    image: seedjeffwan/mnist_tensorflow_keras:1.13.1
    command:
      - "python"
      - "mnist.py"
      - "--model_export_path"
      - "s3://eks-ml-data/mnist/tf_saved_model"
      - "--model_summary_path"
      - "s3://eks-ml-data/mnist/tf_summary"
      - "--epochs"
      - "40"
    env:
      - name: AWS_ACCESS_KEY_ID
        valueFrom:
          secretKeyRef:
            name: aws-secret
            key: AWS_ACCESS_KEY_ID
      - name: AWS_SECRET_ACCESS_KEY
        valueFrom:
          secretKeyRef:
            name: aws-secret
            key: AWS_SECRET_ACCESS_KEY
      - name: AWS_REGION
        value: us-west-2