目前在训练手绘图形的分类,10个大类,样本量差了20倍,精确度上不去;查询资料使用加权交叉熵成功解决,目前精确度可到98%+
## 优化目标使用加权交叉熵
class_weights = torch.FloatTensor(class_weights).to(device)
criterion = nn.CrossEntropyLoss(weight=class_weights)
目前在训练手绘图形的分类,10个大类,样本量差了20倍,精确度上不去;查询资料使用加权交叉熵成功解决,目前精确度可到98%+
## 优化目标使用加权交叉熵
class_weights = torch.FloatTensor(class_weights).to(device)
criterion = nn.CrossEntropyLoss(weight=class_weights)