Fewshot_Detection RuntimeError: shape '[32, 15, 845]' is invalid for input of size 540800

Hi, Thanks for your work , but when I run the code , a error happened, ** **Traceback (most recent call last): File "train_meta.py", line 352, in train(epoch) File "train_meta.py", line 197, in train loss,recall_rate,conf_proposal,loss_bbox,loss_conf,loss_cls = region_loss(output, target) File "/home/plc319/anaconda3/envs/ct/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/home/plc319/桌面/Few-shot-Object-Detection-via-Feature-Reweighting-main/models/layers/region_loss.py", line 296, in forward cls = cls.view(bs, cs, nA * nC * nH * nW).transpose(1, 2).contiguous().view(bs * nA * nC * nH * nW, cs) RuntimeError: shape '[32, 15, 845]' is invalid for input of size 540800 ** can you tell me how to solve the problem **

Apr 28 '21 13:04 Zyt-wenwne

I have the same problem while training base model, and my enviroment is cuda10.1 torch0.3.1 torchvision0.5.0, is there anyone could help us?

Apr 29 '21 03:04 whattoshow

Hi, Thanks for your work , but when I run the code , a error happened, ** **Traceback (most recent call last): File "train_meta.py", line 352, in train(epoch) File "train_meta.py", line 197, in train loss,recall_rate,conf_proposal,loss_bbox,loss_conf,loss_cls = region_loss(output, target) File "/home/plc319/anaconda3/envs/ct/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/home/plc319/桌面/Few-shot-Object-Detection-via-Feature-Reweighting-main/models/layers/region_loss.py", line 296, in forward cls = cls.view(bs, cs, nA * nC * nH * nW).transpose(1, 2).contiguous().view(bs * nA * nC * nH * nW, cs) RuntimeError: shape '[32, 15, 845]' is invalid for input of size 540800 ** can you tell me how to solve the problem **

您好我遇到了同样的问题您解决了吗？有偿