swa.yaml « callbacks « conf « training - text-recognizer.git - A neural network that reads handwritten paragraphs.

summary refs log tree commit diff

path: root/training/conf/callbacks/swa.yaml

blob: 279ca69da0d7e958dc91b12b9bec36b22a7194c7 (plain)

type: StochasticWeightAveraging
args:
    swa_epoch_start: 0.8
    swa_lrs: 0.05
    annealing_epochs: 10
    annealing_strategy: cos
    device: null