这个屌丝很懒，什么也没留下！

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

作者：黑客灵魂 | 2024-07-21 10:48:04

assertion n classes

报错原因

这是由于求交叉熵函数在计算时遇到了类别错误的问题，即不满足t >= 0 && t < n_classes条件。

本次在做七分类任务时的标签为 (1, 2, ... , 7) 导致出现这个报错，把标签改为 (0, 1, ... , 6) 后报错解决。

Dataset类中返回标签值时减一。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/黑客灵魂/article/detail/860554