prediction problem

[vcvishal]

when i tried to predict this error occurred

load model... predicting... predict.py:143: UserWarning: volatile was removed and now has no effect. Use with torch.no_grad(): instead. img = Variable(img[None,:,:,:],volatile=True) D:\miniconda\lib\site-packages\torch\nn\functional.py:1332: UserWarning: nn.functional.sigmoid is deprecated. Use torch.sigmoid instead. warnings.warn("nn.functional.sigmoid is deprecated. Use torch.sigmoid instead.") Traceback (most recent call last): File "predict.py", line 174, in result = predict_gpu(model,image_name) File "predict.py", line 148, in predict_gpu boxes,cls_indexs,probs = decoder(pred) File "predict.py", line 89, in decoder cls_indexs = torch.cat(cls_indexs,0) #(n,) RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated

please guide thank you

[ChengJianjia]

Hi，I met the same problem，have you solve it？

[yahsiuhsieh]

@JianJiaCheng Try to use torch.stack() instead.

[TanMeng0]

@vcvishal @JianJiaCheng add： cls_indexs = [item.unsqueeze(0) for item in cls_indexs] before ： cls_indexs = torch.cat(cls_indexs,0)