Add sample.yaml
Signed-off-by: cuidongdong <515783741@qq.com>
This commit is contained in:
parent
eb0996e222
commit
bdeab36439
|
@ -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
|
Loading…
Reference in New Issue