#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

25

26

27

28

29

30

31

            defaults:
  - cityscapes_dataset_params
  - _self_
train_dataset_params:
  transforms:
    - ColorJitterSeg:
        brightness: 0.5
        contrast: 0.5
        saturation: 0.5
    - RandomFlipSeg:
        prob: 0.5
    - RandomRescaleSeg:
        scales: [ 0.5, 2. ]
    - PadShortToCropSizeSeg:
        crop_size: [ 1024, 1024 ]
        fill_mask: 19
    - CropImageAndMaskSeg:
        crop_size: [ 1024, 1024 ]
        mode: random
train_dataloader_params:
  batch_size: 6
val_dataloader_params:
  batch_size: 6

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