#367 fix: Request correct hydra-core

Merged

Ghost merged 1 commits into Deci-AI:master from deci-ai:hotfix/ALG-000_hydra-req

  
    
        
          
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

          
            defaults:
  - imagenet_dataset_params
train_dataset_params:
  root: /data/Imagenet/train
  transforms:
    - RandomResizedCropAndInterpolation:
        size: 224
        interpolation: random
    - RandomHorizontalFlip
    - RandAugmentTransform:
        config_str: rand-m7-mstd0.5
        crop_size: 224
        img_mean: ${dataset_params.img_mean}  # Use default value from imagenet_dataset_params
    - ToTensor
    - Normalize:
        mean: ${dataset_params.img_mean}      # Use default value from imagenet_dataset_params
        std: ${dataset_params.img_std}        # Use default value from imagenet_dataset_params
val_dataset_params:
  root: /data/Imagenet/val
  transforms:
    - Resize:
        size: 236
    - CenterCrop:
        size: 224
    - ToTensor
    - Normalize:
        mean: ${dataset_params.img_mean}
        std: ${dataset_params.img_std}
train_dataloader_params:
  batch_size: 236
  collate_fn:
    _target_: super_gradients.training.datasets.mixup.CollateMixup
    mixup_alpha: 0.2
    cutmix_alpha: 1.0
    label_smoothing: 0.1

          
        
      

  

Tip!

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

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

#367 fix: Request correct hydra-core

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