RuntimeError: result type Byte can‘t be cast to the desired output type Bool

在PointPillar训练的阶段,出现了一个小bug:也就是一个需要bool型但是代码却是一个byte的错,需要做如下的修改

second.pytorch\second\pytorch\models文件夹下的voxelnet.py的911行:

                    opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.byte()

改为

                    opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.bool()

 好了,接下来训练就完美进行了。


版权声明:本文为m0_57213528原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。