mirror of https://github.com/kubeflow/examples.git
37 lines
1.2 KiB
YAML
37 lines
1.2 KiB
YAML
---
|
|
apiVersion: batch/v1
|
|
kind: Job
|
|
metadata:
|
|
name: batch-predict-object-detection
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: batch-predict-object-detection
|
|
image: gcr.io/kubeflow-examples/batch-predict:tf18-gpu
|
|
imagePullPolicy: Always
|
|
args:
|
|
- "--input_file_pattern=gs://kubeflow-examples-data/object-detection-coco/data/object-detection-images.tfrecord"
|
|
- "--input_file_format=tfrecord"
|
|
- "--model_dir=gs://kubeflow-examples-data/object-detection-coco/image_string_model/saved_model"
|
|
- "--batch_size=4"
|
|
- "--output_result_prefix=gs://<your_gcs_bucket>/<your_detection_result_file_name_prefix>"
|
|
- "--output_error_prefix=gs://<your_gcs_bucket>/<your_detection_error_file_name_prefix>"
|
|
env:
|
|
- name: "GOOGLE_APPLICATION_CREDENTIALS"
|
|
value: "/secret/gcp-credentials/key.json"
|
|
volumeMounts:
|
|
- name: gcp-credentials
|
|
readOnly: True
|
|
mountPath: /secret/gcp-credentials
|
|
resources:
|
|
limits:
|
|
nvidia.com/gpu: 1
|
|
restartPolicy: Never
|
|
volumes:
|
|
- name: gcp-credentials
|
|
secret:
|
|
secretName: user-gcp-sa
|
|
|
|
backoffLimit: 1
|