nn.CrossEntropyLoss(),weight set

权重设置
更改样本中每类的不平衡状况

weights=torch.FloatTensor([0.1,0.5,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,1e-10,1e-10,1e-10,1e-10])

criterion=nn.CrossEntropyLoss(weight=weights,ignore_index=255)

你可能感兴趣的:(人工智能,pytorch)