在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()