小尺寸输入训练报错

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4

1.86k stars 585 forks source link

小尺寸输入训练报错 #131

Open AiLoving opened 3 years ago

AiLoving commented 3 years ago

你好，我下载的master，cfg和weight用的yolov4-pacsp-x-mish，推理图片正常，然后训练时候，因为我的是1050显卡，设置了输入尺寸320*320，但是报错： File "D:\PyTorch_YOLOv4-master\models.py", line 197, in forward p = p.view(bs, self.na, self.no, self.ny, self.nx).permute(0, 1, 3, 4, 2).contiguous() # prediction RuntimeError: shape '[1, 3, 6, 40, 40]' is invalid for input of size 8000 请问可能是什么原因呢

AiLoving commented 3 years ago

难道是只能用640*640训练？

WongKinYiu commented 3 years ago

128的倍數都可以

xuhao-anhe commented 3 years ago

难道是只能用640*640训练？

请问你解决了吗