在PyTorch中,我们可以使用torch.nn模块中的各种损失函数来定义损失函数。以下是一些常用的损失函数及其定义方法:
criterion = torch.nn.MSELoss()
criterion = torch.nn.CrossEntropyLoss()
criterion = torch.nn.NLLLoss()
criterion = torch.nn.BCELoss()
criterion = torch.nn.KLDivLoss()
使用时,我们可以在模型训练过程中计算损失并通过优化器来最小化损失函数。例如:
loss = criterion(output, target)
optimizer.zero_grad()
loss.backward()
optimizer.step()