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