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

#604 fix master installation

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-000_fix_master_inastallation
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
  1. defaults:
  2. - default_train_params
  3. max_epochs: 10
  4. initial_lr: 0.03
  5. lr_mode: cosine
  6. cosine_final_lr_ratio: 0
  7. lr_warmup_epochs: 1
  8. warmup_initial_lr: 0
  9. warmup_mode: linear_step
  10. ema: False
  11. loss: cross_entropy
  12. clip_grad_norm: 1
  13. optimizer: SGD
  14. optimizer_params:
  15. weight_decay: 0
  16. momentum: 0.9
  17. train_metrics_list: # metrics for evaluation
  18. - Accuracy
  19. - Top5
  20. valid_metrics_list: # metrics for evaluation
  21. - Accuracy
  22. - Top5
  23. metric_to_watch: Accuracy
  24. greater_metric_to_watch_is_better: True
  25. average_best_models: False
  26. _convert_: all
Discard
Tip!

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