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

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git

Watch 4
Issues 62 Pull Requests 44 Discussions 0

#211 SG-136: Apply ema only on student (KD)

Merged
Louis Dupont merged 1 commits into Deci-AI:master from deci-ai:feature/SG-136_use_ema_only_on_kd_student
Conversation 0 Files changed 5 Commits 1
File Compare
super-gradients
/
src
/
super_gradients
/
recipes
/
dataset_params
/
coco_detection_dataset_params.yaml
View left file View right file
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 
  1. batch_size: 64 # batch size for trainset in CoCoDetectionDatasetInterface
    2. 

  2. val_batch_size: 64 # batch size for valset in CoCoDetectionDatasetInterface
    3. 

  3. val_image_size: 320 # image size for valset in CoCoDetectionDatasetInterface
    4. 

  4. train_image_size: 320 # image size for trainset in CoCoDetectionDatasetInterface
    5. 

  5. 

  6. val_collate_fn: # collate function for valset in CoCoDetectionDatasetInterface
    7. 

  7.   _target_: super_gradients.training.utils.utils.load_func
    8. 

  8.   dotpath: super_gradients.training.utils.detection_utils.base_detection_collate_fn
    9. 

  9. 

  10. train_collate_fn: # collate function for trainset in CoCoDetectionDatasetInterface
    11. 

  11.   _target_: super_gradients.training.utils.utils.load_func
    12. 

  12.   dotpath: super_gradients.training.utils.detection_utils.base_detection_collate_fn
    13. 

  13. 

  14. val_sample_loading_method: "default"
    15. 

  15. dataset_hyper_param:
    16. 

  16.   hsv_h: 0.015  # IMAGE HSV-Hue AUGMENTATION (fraction)
    17. 

  17.   hsv_s: 0.7  # IMAGE HSV-Saturation AUGMENTATION (fraction)
    18. 

  18.   hsv_v: 0.4  # IMAGE HSV-Value AUGMENTATION (fraction)
    19. 

  19.   degrees: 0.0  # IMAGE ROTATION (+/- deg)
    20. 

  20.   translate: 0.1  # IMAGE TRANSLATION (+/- fraction)
    21. 

  21.   scale: 0.5  # IMAGE SCALE (+/- gain)
    22. 

  22.   shear: 0.0  # IMAGE SHEAR (+/- deg)
    23. 

  23.   mixup: 0.0  # IMAGE MIXUP AUGMENTATION (fraction)
    24. 

  24. 

  25. _convert_: all
    26.
Discard
Prev
Next
Tip!

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