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

#396 Trainer constructor cleanup

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-266_clean_trainer_ctor
Conversation 0 Files changed 49 Commits 1
File Compare
super-gradients
/
src
/
super_gradients
/
recipes
/
dataset_params
/
coco_segmentation_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
47
48
49
50
51
52
53 
  1. train_dataset_params:
    2. 

  2.   root_dir: /data/coco/
    3. 

  3.   list_file: instances_train2017.json
    4. 

  4.   samples_sub_directory: images/train2017
    5. 

  5.   targets_sub_directory: annotations
    6. 

  6.   dataset_classes_inclusion_tuples_list:
    7. 

  7.     _target_: super_gradients.training.utils.segmentation_utils.coco_sub_classes_inclusion_tuples_list
    8. 

  8.   cache_labels: False
    9. 

  9.   cache_images: False
    10. 

  10.   transforms:
    11. 

  11.     # for more options see common.factories.transforms_factory.py
    12. 

  12.     - RandomFlipSeg:
    13. 

  13.         prob: 0.5
    14. 

  14. 

  15.     - RescaleSeg: # consider removing this step
    16. 

  16.         long_size: 608
    17. 

  17. 

  18.     - RandomRescaleSeg:
    19. 

  19.         scales: [ 0.5, 2.0 ]
    20. 

  20. 

  21.     - PadShortToCropSizeSeg:
    22. 

  22.         crop_size: 512
    23. 

  23. 

  24.     - CropImageAndMaskSeg:
    25. 

  25.         crop_size: 512
    26. 

  26.         mode: random
    27. 

  27. 

  28. val_dataset_params:
    29. 

  29.   root_dir: /data/coco/
    30. 

  30.   list_file: instances_val2017.json
    31. 

  31.   samples_sub_directory: images/val2017
    32. 

  32.   targets_sub_directory: annotations
    33. 

  33.   dataset_classes_inclusion_tuples_list:
    34. 

  34.     _target_: super_gradients.training.utils.segmentation_utils.coco_sub_classes_inclusion_tuples_list
    35. 

  35.   cache_labels: False
    36. 

  36.   cache_images: False
    37. 

  37.   transforms:
    38. 

  38.     - RescaleSeg:
    39. 

  39.         short_size: 512
    40. 

  40. 

  41.     - CropImageAndMaskSeg:
    42. 

  42.         crop_size: 512
    43. 

  43.         mode: center
    44. 

  44. 

  45. train_dataloader_params:
    46. 

  46.   batch_size: 8
    47. 

  47.   num_workers: 8
    48. 

  48.   drop_last: True                 # drop the last incomplete batch, if dataset size is not divisible by the batch size
    49. 

  49. 

  50. val_dataloader_params:
    51. 

  51.   batch_size: 24
    52. 

  52.   num_workers: 8
    53. 

  53.   drop_last: False
    54.
Discard
Prev
Next
Tip!

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