nll_loss_forward_reduce_cuda_kernel_2d: Assertion `t >= 0 && t < n__classes` failed.

nll_loss_forward_reduce_cuda_kernel_2d: Assertion `t >= 0 && t < n__classes` failed._../aten/src/aten/native/cuda/loss.cu:242: nll_loss-CSDN博客

参考如上,第二种方法解决的,我在数据量化时,将label设为了1-8,自此报各种奇怪错误。

总结:label需要从0开始,原因不知。

你可能感兴趣的:(机器学习,人工智能,linux,运维)