RuntimeError: cannot unsqueeze empty tensor

[Shualite]

Traceback (most recent call last): File "eval_all.py", line 380, in <module> main() File "eval_all.py", line 347, in main out, seg_pred, seg_map =net(img) File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__ result = self.forward(*input, **kwargs) File "/root/fsy_SceneTextRec/Docker-pytorch-0.3/corner-master/dssd.py", line 113, in forward output = self.detect(loc.view(loc.size(0), -1, 4, 4), self.softmax(conf.view(-1, self.num_classes)), self.priors, F.sigmoid(seg_view), F.sigmoid(seg_pred)) File "/root/fsy_SceneTextRec/Docker-pytorch-0.3/corner-master/layers/functions/detection.py", line 50, in forward temp.append(torch.cat((scores[ids[:count]].unsqueeze(1),

I also find that this code can only run in the pytorch environment under pytorch0.4, otherwise we need to change the c code in the roi_pooling module.
I meet this error above when i run on pytorch0.4.1 and run the eval_all.py

[Shualite]

@lvpengyuan

[lhn217]

@lvpengyuan

do you resolve this problem now?

[zhangchurong]

+1