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

imagenet_regnetY800_train_params.yaml 663 B

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
  1. defaults:
  2. - default_train_params
  3. max_epochs: 100
  4. lr_mode: cosine
  5. initial_lr: 0.05
  6. optimizer: SGD
  7. optimizer_params:
  8. weight_decay: 5e-5
  9. "momentum": 0.9
  10. loss: cross_entropy
  11. loss_logging_items_names: [Loss]
  12. metric_to_watch: Accuracy
  13. greater_metric_to_watch_is_better: True
  14. train_metrics_list: # metrics for evaluation
  15. - _target_: super_gradients.training.metrics.Accuracy
  16. - _target_: super_gradients.training.metrics.Top5
  17. valid_metrics_list: # metrics for evaluation
  18. - _target_: super_gradients.training.metrics.Accuracy
  19. - _target_: super_gradients.training.metrics.Top5
  20. _convert_: all
Tip!

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

Comments

Loading...