Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

dvc.yaml 1.1 KB

You have to be logged in to leave a comment. Sign In
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
30
31
32
33
34
  1. stages:
  2. train:
  3. cmd: python model/src/train.py 2>&1 | tee train.log
  4. deps:
  5. - datasets/samples/preprocessed/
  6. - model/src/train.py
  7. - datasets/trainlist.txt
  8. - params.yaml
  9. outs:
  10. - model/train/${model_name}/checkpoint.h5:
  11. checkpoint: true
  12. - model/train/${model_name}/best.h5:
  13. checkpoint: true
  14. metrics:
  15. - model/train/${model_name}/best.json:
  16. checkpoint: true
  17. - model/train/${model_name}/checkpoint.json:
  18. checkpoint: true
  19. plots:
  20. - model/train/${model_name}/plot.csv:
  21. cache: false
  22. evaluate:
  23. cmd: python model/src/evaluator.py model/train/${model_name}/best.h5 params.yaml datasets/testlist.txt --output model/eval/${model_name}/ 2>&1 | tee eval.log
  24. deps:
  25. - model/train/${model_name}/best.h5
  26. - model/src/evaluator.py
  27. - datasets/testlist.txt
  28. - params.yaml
  29. metrics:
  30. - model/eval/${model_name}/metrics.json:
  31. cache: false
  32. - model/eval/${model_name}/result.csv:
  33. cache: false
Tip!

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

Comments

Loading...