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

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

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

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

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

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

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