#875 Feature/sg 761 yolo nas

Merged

Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-761-yolo-nas

  
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

            # STDC Base training params
defaults:
  - training_hyperparams: cityscapes_default_train_params
  - dataset_params: cityscapes_dataset_params
  - checkpoint_params: default_checkpoint_params
  - _self_
  - variable_setup
train_dataloader: cityscapes_train
val_dataloader: cityscapes_val
data_loader_num_workers: 10
arch_params:
  num_classes: 19
  use_aux_heads: True
checkpoint_params:
  checkpoint_path:
  load_backbone: True
  load_weights_only: True
  strict_load: no_key_matching
architecture: stdc1_seg
experiment_name: ${architecture}_cityscapes
training_hyperparams:
  sync_bn: True
multi_gpu: DDP

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

#875 Feature/sg 761 yolo nas

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