如何处理device-side assert triggered报错？

1、1. pytorch如何处理device-side assert triggered？

2、2. 遇到问题，第一时间百度，问题可能是由于标签数超过了分类数

3、3. 解决方法给出的是将标签进行上下线设置

4、4. 但是我添加后仍然报错

5、5. 仔细检查后终于发现了问题

6、6. 由于网络结构是复制过来的之前的程序是2分类，而现在是6分类问题，只需要改为6即可