summaryrefslogtreecommitdiff
path: root/training/conf/lr_scheduler/one_cycle.yaml
blob: 60a6f27a02181806de91933a7a8574eda9ac437e (plain)
1
2
3
4
5
6
7
8
type: OneCycleLR
args:
  interval: step
  max_lr: 1.0e-3
  three_phase: true
  epochs: 64
  steps_per_epoch: 633 # num_samples / batch_size
monitor: val_loss