今天在用Pytorch跑代码时碰到了一个错误——Assertion ‘cur_target >= 0 && cur_target < n_classes’ failed. at /tmp/pip-req-build-x5rxronk/aten/src/THNN/generic/ClassNLLCriterion.c: 97,如下图所示。
从错误提示信息中可以很容易看出是loss函数计算时,预测类别和目标类别不一致造成的错误,修改即可。
今天在用Pytorch跑代码时碰到了一个错误——Assertion ‘cur_target >= 0 && cur_target < n_classes’ failed. at /tmp/pip-req-build-x5rxronk/aten/src/THNN/generic/ClassNLLCriterion.c: 97,如下图所示。
从错误提示信息中可以很容易看出是loss函数计算时,预测类别和目标类别不一致造成的错误,修改即可。