training/conf/lr_scheduler/one_cycle.yaml


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

_target_: torch.optim.lr_scheduler.OneCycleLR
max_lr: 1.0e-3
total_steps: null
epochs: 512
steps_per_epoch: 4992
pct_start: 0.3
anneal_strategy: cos
cycle_momentum: true
base_momentum: 0.85
max_momentum: 0.95
div_factor: 25.0
final_div_factor: 10000.0
three_phase: true
last_epoch: -1
verbose: false

# Non-class arguments
interval: step
monitor: val/loss