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

permute op may have more than 4 dims #110

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

bailiqun
Copy link

for example last layers of SSD need to permute tensor of 4 dims, but yolov5 need to permute tensor is 5. master version must be 4 dim tensor, may occur convert error.

@yanzhou-li
Copy link

yanzhou-li commented Jun 28, 2021

for example last layers of SSD need to permute tensor of 4 dims, but yolov5 need to permute tensor is 5. master version must be 4 dim tensor, may occur convert error.
@bailiqun, @xxradon
hi,i want to transform the code out = x.permute(0,1,2,4,5,3).contiguous().i changed the code Follow your instructions. but it is still an error. as follow:
file "./bisenetv2.py", line 58, in forward shuffle_out = input_view.permute(0, 1, 4, 2, 5, 3).contiguous() File "./pytorch_to_caffe.py", line 619, in _permute bottom=[log.blobs(input)], top=[log.blobs(x)]) File "./Caffe/layer_param.py", line 33, in __init__ self.bottom.extend(bottom) TypeError: None has type NoneType, but expected one of: bytes, unicode
could you tell how to resolve it? by the way, it doesn't seem to have been modified in the code setattr(param,key,param.Value(value)),where layer_param.py,183
looking forward to your reply!

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

Successfully merging this pull request may close these issues.

2 participants