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

#869 Add DagsHub Logger to Super Gradients

Merged
Ghost merged 1 commits into Deci-AI:master from timho102003:dagshub_logger
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
  1. depth_mult:
  2. width_mult:
  3. num_classes: 80
  4. backbone:
  5. layers: [ 3, 6, 6, 3 ] # Backbone's structure
  6. channels: [ 64, 128, 256, 512, 1024 ] # Number of outputs channels for stem and consecutive feature maps
  7. activation: silu
  8. return_idx: [ 1, 2, 3 ] # Indexes of feature maps to output, indiced 1,2,3 correspond to feature maps of stride 8,16,32
  9. use_large_stem: True # If True, uses 3 conv+bn+act instead of 2 in stem blocks
  10. use_alpha: False # If True, enables additional learnable weighting parameter for 1x1 branch in RepVGGBlock
  11. pretrained_weights:
  12. neck:
  13. in_channels: [256, 512, 1024]
  14. out_channels: [768, 384, 192]
  15. activation: silu
  16. block_num: 3
  17. stage_num: 1
  18. spp: True
  19. head:
  20. in_channels: [768, 384, 192]
  21. activation: silu
  22. fpn_strides: [32, 16, 8]
  23. grid_cell_scale: 5.0
  24. grid_cell_offset: 0.5
  25. reg_max: 16 # Number of bins for size prediction
  26. eval_size: # Size of the image for evaluation. Setting this value can be beneficial for inference speed since anchors will not be regenerated for each forward call.
Discard
Tip!

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