Add sample.yaml

Signed-off-by: cuidongdong <515783741@qq.com>
This commit is contained in:
cuidongdong 2024-07-02 09:07:48 +08:00
parent eb0996e222
commit bdeab36439
1 changed files with 52 additions and 0 deletions

52
sample.yaml Normal file
View File

@ -0,0 +1,52 @@
# PIPELINE DEFINITION
# Name: sample-pipeline
# Inputs:
# batch_size: int
components:
comp-train:
executorLabel: exec-train
inputDefinitions:
parameters:
batch_size:
parameterType: NUMBER_INTEGER
outputDefinitions:
artifacts:
model:
artifactType:
schemaTitle: system.Model
schemaVersion: 0.0.1
parameters:
batch_size:
parameterType: NUMBER_INTEGER
deploymentSpec:
executors:
exec-train:
container:
args:
- echo 'I am a model' > {{$.outputs.artifacts['model'].path}}
command:
- sh
- -c
image: python:3.10
pipelineInfo:
name: sample-pipeline
root:
dag:
tasks:
train:
cachingOptions:
enableCache: true
componentRef:
name: comp-train
inputs:
parameters:
batch_size:
componentInputParameter: batch_size
taskInfo:
name: train
inputDefinitions:
parameters:
batch_size:
parameterType: NUMBER_INTEGER
schemaVersion: 2.1.0
sdkVersion: kfp-2.4.0