RuntimeError: dimension out of range (expected to be in range of [-2, 1], but got 2)

[ycp007]

When I input 21.bmp, I got an error:
please input file name:21.bmp
Traceback (most recent call last):
File "demo.py", line 23, in
crnnRec(model,converter,img,text_recs)
File "/home/ycp/sceneReco-master/crnnport.py", line 78, in crnnRec
preds = preds.squeeze(2)
File "/usr/local/lib/python2.7/dist-packages/torch/autograd/variable.py", line 750, in squeeze
return Squeeze.apply(self, dim)
File "/usr/local/lib/python2.7/dist-packages/torch/autograd/_functions/tensor.py", line 378, in forward
result = input.squeeze(dim)
RuntimeError: dimension out of range (expected to be in range of [-2, 1], but got 2)

[bear63]

#preds = preds.squeeze(2)
delete preds = preds.squeeze(2) I also encountered this problem on different computers

[liurongfang]

@bear63 thank you,this is ok