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

config.yaml 1.0 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
  1. artifacts_roots: artifact
  2. data_ingestion:
  3. root_dir: artifact/data_ingestion
  4. source_URL: https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_5340.zip
  5. local_data_file: artifact/data_ingestion/data.zip
  6. unzip_dir: artifact/data_ingestion
  7. prepare_base_model:
  8. root_dir: artifact/prepare_base_model
  9. base_model_path: artifact/prepare_base_model/base_model.h5
  10. updated_base_model_path: artifact/prepare_base_model/base_model_updated.h5
  11. prepare_callbacks:
  12. root_dir: artifact/prepare_callbacks
  13. tensorboard_root_log_dir: artifact/prepare_callbacks/tensorboard_log_dir
  14. checkpoint_model_filepath: artifact/prepare_callbacks/checkpoint_dir/model.h5
  15. training:
  16. root_dir: artifact/training
  17. trained_model_path: artifact/training/model.h5
  18. evaluation:
  19. root_dir: artifact/evaluation
  20. score_file_path: artifact/evaluation/scores.json
  21. params:
  22. AUGMENTATION: True
  23. IMAGE_SIZE: [224, 224, 3]
  24. BATCH_SIZE: 16
  25. INCLUDE_TOP: False
  26. EPOCHS: 1
  27. CLASSES: 2
  28. WEIGHTS: imagenet
  29. LEARNING_RATE: 0.01
Tip!

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

Comments

Loading...