Hi, I changed the image size in Cfg file of the YOLOv4 this repo to
Cfg.width = 224
Cfg.height = 224
I got the below error. I am not able to resolve it. Can you help me to find out the problem.
Epoch 1/3: 0%| | 0/2 [00:03<?, ?img/s]
Traceback (most recent call last):
File "train.py", line 634, in <module>
train(model=model,
File "train.py", line 383, in train
loss, loss_xy, loss_wh, loss_obj, loss_cls, loss_l2 = criterion(bboxes_pred, bboxes)
File "C:\ProgramData\Anaconda3\envs\yolov4t\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "train.py", line 249, in forward
pred[..., 0] += self.grid_x[output_id]
RuntimeError: The size of tensor a (28) must match the size of tensor b (76) at non-singleton dimension 3