How to solve this runtime error problem?

ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"

https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/

2.32k stars 724 forks source link

How to solve this runtime error problem? #60

Open sunshine-lcc opened 4 years ago

sunshine-lcc commented 4 years ago

I met the problem when I ran the instruction "python3 detect.py --images imgs --det det" What should I do to solve this problem and why it happened?

Traceback (most recent call last): File "detect.py", line 234, in output = torch.cat((output,prediction)) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 86 and 8 in dimension 1 at /opt/conda/conda-bld/pytorch_1544176307774/work/aten/src/THC/generic/THCTensorMath.cu:83

lilxmx commented 2 years ago

I met the same one， have you resolve it?