_target_: torch.optim.AdamW lr: 1e-5 weight_decay: 0.01 fused: False