machine-learning/convolutional-neural-network/dvc.yaml
---
stages:
prepare:
cmd: tar -x -f data/raw/cifar-10-python.tar.gz -C data/processed
deps:
- data/raw/cifar-10-python.tar.gz
outs:
- data/processed/cifar-10-batches-py/
main:
cmd: python src/main.py
deps:
- data/processed/cifar-10-batches-py/
- src/evaluate.py
- src/main.py
- src/model/
- src/train.py
params:
- lr
- train.epochs
outs:
- output/models/model.pt