blob: 801a01f14366050b305cbcf4b300d1b9c72bee58 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
one_cycle:
_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
|
