Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

yolov3-tiny model image dimensions error #55

Open
TheArheus opened this issue Dec 12, 2019 · 0 comments
Open

yolov3-tiny model image dimensions error #55

TheArheus opened this issue Dec 12, 2019 · 0 comments

Comments

@TheArheus
Copy link

Hello, I was following your tutorial and tried to use tiny version of yolov3 model, but I stucked in step 3. But, the normal version is working fine there.
Error I have is next:

Traceback (most recent call last):
  File "C:/Users/1/PycharmProjects/untitled/yolo.py", line 285, in <module>
    pred = net(in_val, torch.cuda.is_available())
  File "C:\Users\1\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\nn\modules\module.py", line 541, in __call__
    result = self.forward(*input, **kwargs)
  File "C:/Users/1/PycharmProjects/untitled/yolo.py", line 246, in forward
    x = torch.cat((map1, map2), 1)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 18 and 20 in dimension 2 at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensor.cpp:689

How can I fix this problem. Thank you

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant