在做多分类问题的实验时出现了错误,提示:
pytorch CUDA error: device-side assert triggered
搜索发现均说明是标签个数不匹配的问题,仔细查了一下,原来是在对数据集做新的分类后,数据类别为7,而模型没有更改,最后一层还是原来的6个神经元,更改之后就可以了。
pytorch CUDA error: device-side assert triggered
最新推荐文章于 2024-08-23 17:36:30 发布
在做多分类问题的实验时出现了错误,提示:
pytorch CUDA error: device-side assert triggered
搜索发现均说明是标签个数不匹配的问题,仔细查了一下,原来是在对数据集做新的分类后,数据类别为7,而模型没有更改,最后一层还是原来的6个神经元,更改之后就可以了。