type: "adam"
args:
    lr: 5e-4
    betas: [0.9, 0.999]
    eps: 1e-08
    weight_decay: 0.0
    amsgrad: false