From b567905d961bad4eef3fcfd2bc8253826a882536 Mon Sep 17 00:00:00 2001 From: xmyqsh Date: Mon, 6 Jan 2020 23:40:26 +0800 Subject: [PATCH] Use tensor.bool instead of tensor.byte --- det3d/models/bbox_heads/mg_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/det3d/models/bbox_heads/mg_head.py b/det3d/models/bbox_heads/mg_head.py index 8640011..2d7d690 100644 --- a/det3d/models/bbox_heads/mg_head.py +++ b/det3d/models/bbox_heads/mg_head.py @@ -1035,7 +1035,7 @@ def get_task_detections( dir_labels = selected_dir_labels opp_labels = ( (box_preds[..., -1] - self.direction_offset) > 0 - ) ^ dir_labels.byte() + ) ^ dir_labels.bool() box_preds[..., -1] += torch.where( opp_labels, torch.tensor(np.pi).type_as(box_preds),