12 lines
170 B
YAML
12 lines
170 B
YAML
|
stages:
|
||
|
|
train:
|
||
|
|
cmd: python ./train.py 10
|
||
|
|
deps:
|
||
|
|
- Customers.csv
|
||
|
|
- train.py
|
||
|
|
test:
|
||
|
|
cmd: python ./test.py
|
||
|
|
deps:
|
||
|
|
- cifar_net.pth
|
||
|
|
- test.py
|