- 博客(1)
- 收藏
- 关注
原创 device-side assert triggered at /pytorch/aten/src/THC/THCReduceAll.cuh:327
关于device-side assert triggered at /pytorch/aten/src/THC/THCReduceAll.cuh:327 主要是因为交叉熵损失函数的labels出现了越界,labels的值只能为[0,n_class-1]。 之前更改了Maskrcnn的数据集后类的个数应该为50个,但是数据处理的时候直接设置到了50所以一直报错!将label的tensor - 1就完美解决了! 如下代码中注意predicate_labels中的取值即可! self.criterion = nn
2022-03-13 21:25:02 4282
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人