# @package train.optimizer _target_: torch.optim.SGD