From a3afb1200044f96417ee89d3e24ac2ca32dcb6ac Mon Sep 17 00:00:00 2001 From: Dror Ayalon Date: Fri, 1 May 2020 23:51:17 -0400 Subject: [PATCH] bug fix for torch.max operation RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2 'other' in call to _th_max --- model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model.py b/model.py index 9deaedf..43cd2bf 100644 --- a/model.py +++ b/model.py @@ -491,7 +491,8 @@ def detect_objects(self, predicted_locs, predicted_scores, min_score, max_overla # Suppress boxes whose overlaps (with this box) are greater than maximum overlap # Find such boxes and update suppress indices - suppress = torch.max(suppress, overlap[box] > max_overlap) + overlap_above_max = (overlap[box] > max_overlap).byte() + suppress = torch.max(suppress, overlap_above_max) # The max operation retains previously suppressed boxes, like an 'OR' operation # Don't suppress this box, even though it has an overlap of 1 with itself