6 lines
92 B
YAML
6 lines
92 B
YAML
stages:
|
|
train:
|
|
cmd: python train.py --epochs=10
|
|
predict:
|
|
cmd: python predict.py
|