PyTorch使用F.cross_entropy报错Assertion `t ＞= 0 && t ＜ n_classes` failed 和解决RuntimeError: CUDA error: de
和解决RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors…CUDA_LAUNCH_BLOCKING=1)
·
PyTorch使用F.cross_entropy报错Assertion `t ＞= 0 && t ＜ n_classes` failed 和解决RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors...CUDA_LAUNCH_BLOCKING=1
和解决RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors…CUDA_LAUNCH_BLOCKING=1)
第一点
修改网络的（分类任务）的n_class,未修改输出的类别，导致交叉熵loss计算出现错误。
第二点
输入数据用的xml或者csv文件的标签设置有问题
腾讯云面向开发者汇聚海量精品云计算使用和开发经验，营造开放的云计算技术生态圈。
更多推荐
1
0- 0
已为社区贡献2条内容
所有评论(0)