#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

32

33

34

35

36

37

            defaults:
  - default_train_params
max_epochs: 250
lr_updates:
  _target_: numpy.arange
  start: 100
  stop: 250
  step: 50
lr_decay_factor: 0.1
lr_mode: step
lr_warmup_epochs: 0
initial_lr: 0.1
loss: cross_entropy
optimizer: SGD
criterion_params: {}
optimizer_params:
  weight_decay: 1e-4
  momentum: 0.9
metric_to_watch: Accuracy
greater_metric_to_watch_is_better: True
train_metrics_list:                               # metrics for evaluation
  - Accuracy
  - Top5
valid_metrics_list:                               # metrics for evaluation
  - Accuracy
  - Top5
_convert_: all

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