#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

38

39

40

41

42

43

44

            defaults:
  - default_train_params
max_epochs: 450
lr_mode: step
initial_lr: 0.032   # for total batch-size of 512
lr_decay_factor: 0.973
lr_updates:
  _target_: numpy.arange
  start: 2.4
  stop: 450
  step: 2.4
lr_warmup_epochs: 5
optimizer: RMSpropTF
optimizer_params:
  weight_decay: 0.00001
  momentum: 0.9
  alpha: 0.9
  eps: 0.001
loss: cross_entropy
zero_weight_decay_on_bias_and_bn: True
ema: True
ema_params:
  decay: 0.9999
mixed_precision: True
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