首先，必须将权重也转为Tensor的cuda格式；
然后，将该class_weight作为交叉熵函数对应参数的输入值。
class_weight = torch.FloatTensor([0.13859937, 0.5821059, 0.63871904, 2.30220396, 7.1588294, 0]).cuda()
pytorch交叉熵损失函数的weight参数
猜你喜欢
转载自blog.csdn.net/xijuezhu8128/article/details/86594621
今日推荐
周排行