_target_: torch.optim.Adam
_partial_: true
lr: 1e-4
weight_decay: 0.0