Each project can contain a single valohai.yaml file. As your project expands, you may encounter repetitive content within this YAML file.
To address this, we suggest utilizing YAML anchors, aliases, and extensions to define and efficiently reuse duplicated content.
- definitions:
my-inputs: &my_inputs
- name: test-set
keep-directories: suffix
default: s3://onboard-sample/test/*
- name: train-set
keep-directories: suffix
default: s3://onboard-sample/train/*
- step:
name: train-model
image: tensorflow/tensorflow:2.0.1
command:
- pip install valohai-utils numpy
- python ./train.py {parameters}
parameters:
- name: epoch
default: 5
type: integer
inputs: *my_inputs