name: torch.optim.AdamW

kwargs: 
    lr: 0.0001
    betas: [0.9, 0.999]    
    weight_decay: 0.0001
