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.7 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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
  1. # add stages here
  2. # stages:
  3. # stage_name:
  4. # cmd: python src/stage_01______.py
  5. # deps:
  6. # - src/stage_01______.py
  7. # outs:
  8. # - output_data_of_file_path
  9. # add stages here
  10. stages:
  11. get_data:
  12. cmd: python src/stage_01_get_data.py --config=configs/config.yaml
  13. deps:
  14. - src/stage_01_get_data.py
  15. - src/utils/common.py
  16. - configs/config.yaml
  17. outs:
  18. - data/cat
  19. - data/dog
  20. prepare_base_model:
  21. cmd: python src/stage_02_prepare_base_model.py --config=configs/config.yaml --params=params.yaml
  22. deps:
  23. - src/stage_02_prepare_base_model.py
  24. - src/utils/common.py
  25. - src/utils/model.py
  26. - configs/config.yaml
  27. params:
  28. - IMAGE_SIZE
  29. - LEARNING_RATE
  30. outs:
  31. - artifacts/base_model/base_model.h5
  32. - artifacts/base_model/updated_base_model.h5
  33. prepare_callback:
  34. cmd: python src/stage_03_prepare_callback.py --config=configs/config.yaml
  35. deps:
  36. - src/stage_03_prepare_callback.py
  37. - src/utils/common.py
  38. - src/utils/callbacks.py
  39. - configs/config.yaml
  40. - params.yaml
  41. outs:
  42. - artifacts/callbacks/checkpoint_cb.cb
  43. - artifacts/callbacks/tensorboard_cb.cb
  44. train:
  45. cmd: python src/stage_04_train.py --config=configs/config.yaml --params=params.yaml
  46. deps:
  47. - src/stage_04_train.py
  48. - src/utils/common.py
  49. - src/utils/model.py
  50. - src/utils/callbacks.py
  51. - src/utils/data_management.py
  52. - configs/config.yaml
  53. - artifacts/callbacks/checkpoint_cb.cb
  54. - artifacts/callbacks/tensorboard_cb.cb
  55. - artifacts/base_model/updated_base_model.h5
  56. - data/cat
  57. - data/dog
  58. params:
  59. - IMAGE_SIZE
  60. - BATCH_SIZE
  61. - AUGMENTATION
  62. - EPOCHS
  63. outs:
  64. - artifacts/model
  65. - artifacts/tensorboard_log_dir
  66. - artifacts/checkpoints
Tip!

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

Comments

Loading...