如何处理device-side assert triggered报错

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

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

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

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

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

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