search

skyhehe123 / SA-SSD

SA-SSD: Structure Aware Single-stage 3D Object Detection from Point Cloud (CVPR 2020)
492 stars 106 forks source link

RuntimeError #22

Open Morgan-Gan opened 4 years ago

Morgan-Gan commented 4 years ago

Any one know that-- opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.byte() RuntimeError: result type Byte can't be cast to the desired output type Bool

stalkermustang commented 4 years ago

What's pytorch version you use?

asadnorouzi commented 4 years ago

@Morgan-Gan You can solve this problem by applying the following change:

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

I think it's just due to different pytorch versions.

Morgan-Gan commented 4 years ago

What's pytorch version you use?

What's pytorch version you use?

pytorch 1.4.1，it's pytorch versions problem. Thank you for your reply ！

Morgan-Gan commented 4 years ago

@Morgan-Gan You can solve this problem by applying the following change:

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

I think it's just due to different pytorch versions.

You are right, thank you !