一些常见的损失函数调包就行,复杂的得自己写。损失函数中没有可训练的参数,因此通常直接使用torch.nn.functional中的函数即可。例如一些简单的损失函数:
torch.nn.functional
1234
import torch.nn as nncls_criterion = nn.CrossEntropyLoss()dist_criterion = nn.MSELoss() # Use L2 loss functionhinge_criterion = nn.HingeEmbeddingLoss()
还可以参照pytorch-metric-learning这个库
AmazingHao
Engineer
Wuhan, Chongqing, China
文章
34
分类
7
标签
22
开源库
专业应用
实用工具
MathUtilities
算法