PyTorch_YOLO-Family icon indicating copy to clipboard operation
PyTorch_YOLO-Family copied to clipboard

Published 20 hours ago • verified publisher icon yjh0410

RuntimeError: The size of tensor a (45) must match the size of tensor b (46) at non-singleton dimension 2

Open kulkarnikeerti opened this issue 2 years ago • 0 comments

I get this error when RuntimeError: The size of tensor a (45) must match the size of tensor b (46) at non-singleton dimension 2I use (720,1440) as an input image size. But the same model works without any issues when I use (960,1920)as image size. How does this image sizes differs? Both of them are rectangle.

https://github.com/yjh0410/PyTorch_YOLO-Family/blob/a26e9ddd3c2263d6384bee0b0f9ef353775fb281/models/yolo/yolo_nano.py#L204

Can you please tell me where its going wrong. It would be great if you can guide me here. Thanks

kulkarnikeerti avatar Jun 23 '22 11:06 kulkarnikeerti