_target_: torch.nn.NLLLoss