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