Runtime error during model.predict() call

Hi all, I have trained a detecto model to detect rectangles and squares in image. After training, when I call model.predict(image), the following runtime error occurs. Any guidance to get rid of this. Thanks I am using: python 3.6 torch: 1.5.0+cu101

here is the call traceback

3 image = utils.read_image('/color_matching/sample_photo/img.jpg')
----> 4 predictions = model.predict_top([image])

/usr/local/lib/python3.6/dist-packages/torchvision/models/detection/_utils.py in decode(self, rel_codes, boxes)

183         box_sum += val
184         pred_boxes = self.decode_single(
-->185             rel_codes.reshape(box_sum, -1), concat_boxes)
186 
187         return pred_boxes.reshape(box_sum, -1, 4)

RuntimeError: cannot reshape tensor of 0 elements into shape [0, -1] because the unspecified dimension size -1 can be any value and is ambiguous.

    *:  The image has size(500, 500)

[anirudh-g]

I had the same issue with model.predict(). But once I switched to torch 1.4 and torchvision 0.5.0, it was alright. This seems to be limited to the recent releases of pytorch, atleast for me

[alankbi]

Closing due to inactivity - if anyone else comes across this issue, feel free to re-open this or create a new one