42 lines
669 B
YAML
42 lines
669 B
YAML
stages:
|
|
prepare:
|
|
cmd: python prepare.py
|
|
deps:
|
|
- prepare.py
|
|
params:
|
|
- prepare
|
|
outs:
|
|
- data/processed
|
|
train:
|
|
cmd: python train.py
|
|
deps:
|
|
- data/processed
|
|
- train.py
|
|
params:
|
|
- train
|
|
outs:
|
|
- model.pt
|
|
evaluate:
|
|
cmd: python evaluate.py
|
|
deps:
|
|
- data/processed
|
|
- evaluate.py
|
|
- model.pt
|
|
params:
|
|
- evaluate
|
|
params:
|
|
- dvclive/train/params.yaml
|
|
- dvclive/eval/params.yaml
|
|
artifacts:
|
|
resnet50:
|
|
path: model.pt
|
|
type: model
|
|
metrics:
|
|
- dvclive/train/metrics.json
|
|
- dvclive/eval/metrics.json
|
|
plots:
|
|
- dvclive/train/plots/metrics:
|
|
x: step
|
|
- dvclive/eval/plots/metrics:
|
|
x: step
|