#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
step_lr_update_freq: 2.4
initial_lr: 0.016
lr_warmup_epochs: 3
warmup_initial_lr: 1e-6
lr_decay_factor: 0.97
optimizer: RMSpropTF
optimizer_params: 
  momentum: 0.9
  weight_decay: 1e-5
  eps: 0.001
ema: True
ema_params:
  exp_activation: False
  decay: 0.9999
loss: cross_entropy
criterion_params:
  smooth_eps: 0.1
metric_to_watch: Accuracy
greater_metric_to_watch_is_better: True
save_ckpt_epoch_list: [50, 100, 150, 200]
average_best_models: True
mixed_precision: True
zero_weight_decay_on_bias_and_bn: 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