#396 Trainer constructor cleanup

Merged

Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-266_clean_trainer_ctor

  
1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

            defaults:
  - imagenet_dataset_params
train_dataset_params:
  root: /data/Imagenet/train
  transforms:
    - RandomResizedCropAndInterpolation:
        size: 224
        interpolation: random
    - RandomHorizontalFlip
    - RandAugmentTransform:
        config_str: rand-m9-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
    - RandomErase:
        probability: 0.2
        value: random
train_dataloader_params:
  batch_size: 64

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

#396 Trainer constructor cleanup

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