IndexError: tensors used as indices must be long, byte or bool tensors

[alirazaanis-inteon]

Error at line https://github.com/NVIDIA/retinanet-examples/blob/main/odtk/box.py#L299

291 classes = (indices / width / height) % num_classes
292 classes = classes.type(all_cls_head.type())
293
294 # Infer kept bboxes
295 x = indices % width
296 y = (indices / width) % height
297 a = indices / num_classes / height / width
298 box_head = box_head.view(num_anchors, num_boxes, height, width)
299 boxes = box_head[a, :, y, x] ---------------------------------------------------------> IndexError

use '//' instead of '/' to fix the issue, fixed code below;

291 classes = (indices // width // height) % num_classes
292 classes = classes.type(all_cls_head.type())
293
294 # Infer kept bboxes
295 x = indices % width
296 y = (indices // width) % height
297 a = indices // num_classes // height // width
298 box_head = box_head.view(num_anchors, num_boxes, height, width)
299 boxes = box_head[a, :, y, x]