super-gradients/src/super_gradients/recipes/training_hyperparams/imagenet_resnet50_train_params.yaml

defaults:
  - default_train_params

max_epochs: 400
initial_lr: 0.1
lr_mode: cosine
lr_warmup_epochs: 5
ema: False
save_ckpt_epoch_list: [ 50, 100, 150, 200, 300 ]
mixed_precision: True
zero_weight_decay_on_bias_and_bn: True
loss: cross_entropy
train_metrics_list:                               # metrics for evaluation
  - Accuracy
  - Top5

valid_metrics_list:                               # metrics for evaluation
  - Accuracy
  - Top5

metric_to_watch: Accuracy
greater_metric_to_watch_is_better: True

_convert_: all