如何处理device-side assert triggered报错

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

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

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

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

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

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