Run multiple pipeline instances in parallel
Last updated
Was this helpful?
Was this helpful?
- step:
name: preprocess_factory_data
image: python:3.10
command:
- pip install valohai-utils
- python ./preprocess.py
parameters:
- name: factory_id
type: string
default: "factory_001"
- name: quality_threshold
type: float
default: 0.95
inputs:
- name: raw_data
default: s3://data/factories/{parameter:factory_id}/raw/
- step:
name: train_quality_model
image: python:3.10
command:
- pip install valohai-utils
- python ./train.py
parameters:
- name: factory_id
type: string
default: "factory_001"
- name: model_type
type: string
default: "quality_inspector"
inputs:
- name: training_data
optional: true
- pipeline:
name: Factory Quality Pipeline
parameters:
- name: factory_identifier
targets:
- preprocess.parameters.factory_id
- train.parameters.factory_id
default: "factory_001"
nodes:
- name: preprocess
step: preprocess_factory_data
type: execution
- name: train
step: train_quality_model
type: execution
override:
inputs:
- name: training_data
edges:
- [preprocess.output.processed_data*, train.input.training_data]factory_001
factory_002
factory_003- step:
name: train_step
image: python:3.10
command:
- pip install valohai-utils
- python ./train.py
parameters:
- name: factory_id
type: string
default: "factory_001"
inputs:
- name: dataset
default: dataset://{parameter:factory_id}/latest # dataset://factory_001/latest
optional: true- pipeline:
name: Factory ML Pipeline
parameters:
- name: factory_id
targets:
- train.parameters.factory_id
nodes:
- name: train
type: task # Each factory runs hyperparameter tuning
step: train_model