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()

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

你可能感兴趣的:(经验分享,ubuntu,pytorch)