Charlotte_8的博客

原创 PyTorch使用F.cross_entropy报错Assertion `t ＞= 0 && t ＜ n_classes` failed问题记录

这是由于求交叉熵函数在计算时遇到了类别错误的问题，即不满足条件。本次在做七分类任务时的标签为 (1, 2, ... , 7) 导致出现这个报错， 把标签改为 (0, 1, ... , 6) 后报错解决。