Spaces:
Runtime error
Runtime error
File size: 720 Bytes
beb5662 53f077b beb5662 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
stages:
train:
cmd: docker-compose run --rm train
deps:
- docker-compose.yaml
- src/train_optuna_callbacks.py
- src/create_artifacts.py
- data
outs:
- checkpoints/best-checkpoint.ckpt
- checkpoints/train_done.flag
# eval:
# cmd: docker-compose run --rm eval
# deps:
# - docker-compose.yaml
# - src/train.py
# - checkpoints/best-checkpoint.ckpt
# - checkpoints/train_done.flag
# inference:
# cmd: docker-compose run --rm inference
# deps:
# - docker-compose.yaml
# - src/infer.py
# - checkpoints/best-checkpoint.ckpt
# - checkpoints/train_done.flag
# outs:
# - artifacts/image_prediction.png
|