源代码为:
loss = nn.functional.cross_entropy(x, target, ignore_index=ignore_index, weight=loss_weight)
target原先为Tensor类型,指定target为long类型即可。
把target改为target.long()
loss = nn.functional.cross_entropy(x, target.long(), ignore_index=ignore_index, weight=loss_weight)