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