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

#1001 Bug/sg 861 decouple qat from train from config

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:bug/SG-861_decouple_qat_from_train_from_config
Conversation 0 Files changed 11 Commits 1
File Compare
super-gradients
/
src
/
super_gradients
/
recipes
/
dataset_params
/
supervisely_persons_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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46 
  1. dataset_dir: /data/supervisely-persons
    2. 

  2. batch_size: 8
    3. 

  3. 

  4. train_dataset_params:
    5. 

  5.   root_dir: ${..dataset_dir}
    6. 

  6.   list_file: train.csv
    7. 

  7.   cache_labels: False
    8. 

  8.   cache_images: False
    9. 

  9.   transforms:
    10. 

  10.     - SegRandomRescale:
    11. 

  11.         scales: [ 0.25, 1. ]
    12. 

  12.     - SegColorJitter:
    13. 

  13.         brightness: 0.5
    14. 

  14.         contrast: 0.5
    15. 

  15.         saturation: 0.5
    16. 

  16.     - SegRandomFlip:
    17. 

  17.         prob: 0.5
    18. 

  18.     - SegPadShortToCropSize:
    19. 

  19.         crop_size: [ 320, 480 ]
    20. 

  20.         fill_mask: 0
    21. 

  21.     - SegCropImageAndMask:
    22. 

  22.         crop_size: [ 320, 480 ]
    23. 

  23.         mode: random
    24. 

  24. 

  25. val_dataset_params:
    26. 

  26.   root_dir: ${..dataset_dir}
    27. 

  27.   list_file: val.csv
    28. 

  28.   cache_labels: False
    29. 

  29.   cache_images: False
    30. 

  30.   transforms:
    31. 

  31.     - SegResize:
    32. 

  32.         h: 480
    33. 

  33.         w: 320
    34. 

  34. 

  35. train_dataloader_params:
    36. 

  36.   dataset: SuperviselyPersonsDataset
    37. 

  37.   shuffle: True
    38. 

  38.   batch_size: ${..batch_size}
    39. 

  39.   drop_last: True
    40. 

  40. 

  41. val_dataloader_params:
    42. 

  42.   dataset: SuperviselyPersonsDataset
    43. 

  43.   batch_size: ${..batch_size}
    44. 

  44.   drop_last: False
    45. 

  45. 

  46. _convert_: all
    47.
Discard
Prev
Next
Tip!

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