_target_: torch.optim.Adam _partial_: true lr: 0.0001 weight_decay: 0.0