5 lines
111 B
YAML
5 lines
111 B
YAML
|
stages:
|
||
|
|
train:
|
||
|
|
cmd: python train.py --train=true --save_model=true
|
||
|
|
test:
|
||
|
|
cmd: python train_model.py
|