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

#875 Feature/sg 761 yolo nas

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-761-yolo-nas
Conversation 0 Files changed 60 Commits 1
File Compare
super-gradients
/
src
/
super_gradients
/
recipes
/
dataset_params
/
cifar10_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
54
55 
  1. batch_size: 256 # batch size for trainset
    2. 

  2. val_batch_size: 512 # batch size for valset in DatasetInterface
    3. 

  3. 

  4. # TODO: REMOVE ABOVE, HERE FOR COMPATIBILITY UNTIL WE REMOVE DATASET_INTERFACE
    5. 

  5. 

  6. train_dataset_params:
    7. 

  7.   root: ./data/cifar10
    8. 

  8.   train: True
    9. 

  9.   transforms:
    10. 

  10.     - RandomCrop:
    11. 

  11.         size: 32
    12. 

  12.         padding: 4
    13. 

  13.     - RandomHorizontalFlip
    14. 

  14.     - ToTensor
    15. 

  15.     - Normalize:
    16. 

  16.         mean:
    17. 

  17.           - 0.4914
    18. 

  18.           - 0.4822
    19. 

  19.           - 0.4465
    20. 

  20.         std:
    21. 

  21.           - 0.2023
    22. 

  22.           - 0.1994
    23. 

  23.           - 0.2010
    24. 

  24.   target_transform: null
    25. 

  25.   download: True
    26. 

  26. 

  27. train_dataloader_params:
    28. 

  28.   shuffle: True
    29. 

  29.   batch_size: 256
    30. 

  30.   num_workers: 8
    31. 

  31.   drop_last: False
    32. 

  32.   pin_memory: True
    33. 

  33. 

  34. val_dataset_params:
    35. 

  35.   root: ./data/cifar10
    36. 

  36.   train: False
    37. 

  37.   transforms:
    38. 

  38.     - ToTensor
    39. 

  39.     - Normalize:
    40. 

  40.         mean:
    41. 

  41.           - 0.4914
    42. 

  42.           - 0.4822
    43. 

  43.           - 0.4465
    44. 

  44.         std:
    45. 

  45.           - 0.2023
    46. 

  46.           - 0.1994
    47. 

  47.           - 0.2010
    48. 

  48.   target_transform: null
    49. 

  49.   download: True
    50. 

  50. 

  51. val_dataloader_params:
    52. 

  52.   batch_size: 512
    53. 

  53.   num_workers: 8
    54. 

  54.   drop_last: False
    55. 

  55.   pin_memory: True
    56.
Discard
Prev
Next
Tip!

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