dvc.yaml 1009 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. stages:
  2. process_data:
  3. cmd: python3 src/code/make_dataset.py src/data/raw/nyu_depth_v2_labeled.mat src/data/raw/splits.mat
  4. src/data/processed
  5. deps:
  6. - src/code/make_dataset.py
  7. - src/data/raw/nyu_depth_v2_labeled.mat
  8. - src/data/raw/splits.mat
  9. outs:
  10. - src/data/processed/
  11. train:
  12. cmd: python3 src/code/training.py src/data/processed/train
  13. deps:
  14. - src/code/custom_data_loading.py
  15. - src/code/params.yml
  16. - src/code/training.py
  17. - src/data/processed/train
  18. outs:
  19. - src/models/
  20. - logs/train_params.yml:
  21. cache: false
  22. metrics:
  23. - logs/train_metrics.csv:
  24. cache: false
  25. eval:
  26. cmd: python3 src/code/eval.py src/data/processed/test
  27. deps:
  28. - src/code/params.yml
  29. - src/code/custom_data_loading.py
  30. - src/code/eval_metric_calculation.py
  31. - src/code/eval.py
  32. - src/models/model.pth
  33. - src/data/processed/test
  34. outs:
  35. - src/eval/examples/
  36. metrics:
  37. - logs/test_metrics.csv:
  38. cache: false
Tip!

Press p or to see the previous file or, n or to see the next file