53 lines
1.1 KiB
YAML
53 lines
1.1 KiB
YAML
|
# 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
|