报错cuda divice error,
实际上是target中有有负数
模型换到cpu上跑,一般真正的错误就能显示出来了了,或者距离真正错误更近了/。
ret = torch._C._nn.nll_loss2d(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
IndexError: Target -1 is out of bounds.
报错cuda divice error,
实际上是target中有有负数
模型换到cpu上跑,一般真正的错误就能显示出来了了,或者距离真正错误更近了/。
ret = torch._C._nn.nll_loss2d(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
IndexError: Target -1 is out of bounds.