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

prune dla34 backbone #156

Closed
shuizhilinxin opened this issue Apr 20, 2023 · 4 comments
Closed

prune dla34 backbone #156

shuizhilinxin opened this issue Apr 20, 2023 · 4 comments

Comments

@shuizhilinxin
Copy link

Hello, thank you very much for your work. I am trying to prune the dla34 model, but I have encountered a problem with
tensor not matching at non-singleton dimension 0
After investigating the reason, I found that there is a convolution self.project after a downsample in the tree block of the dla34 model. It seems that the build dependency did not track self.project, which caused the tensor not match issue.

Can you please tell me how to solve this problem?

@VainF
Copy link
Owner

VainF commented Apr 20, 2023

Hi @shuizhilinxin. I was wondering if you could kindly share the GitHub repository for the DLA34 model with me. I'm not particularly familiar with this model. Thank you in advance.

@shuizhilinxin
Copy link
Author

shuizhilinxin commented Apr 21, 2023

谢谢大佬的回复,这个链接https://github.com/xingyizhou/CenterNet/blob/2b7692c377c6686fb35e473dac2de6105eed62c6/src/lib/models/networks/dlav0.py,使用def dla34(pretrained, **kwargs): # DLA-34,在Tree block的self.project处failed

@janthmueller
Copy link

谢谢大佬的回复,这个链接https://github.com/xingyizhou/CenterNet/blob/2b7692c377c6686fb35e473dac2de6105eed62c6/src/lib/models/networks/dlav0.py,使用def dla34(pretrained, **kwargs): # DLA-34,在Tree block的self.project处failed

Were you able to find and fix the problem? Could you possibly share your results? I'm not too familiar with the Torch-Vision repo, but I have experience with CenterNet and DLA. It seems like the unused layers are a problem: ucbdrive/dla#23 (comment)

@lebron-2016
Copy link

谢谢大佬的回复,这个链接https://github.com/xingyizhou/CenterNet/blob/2b7692c377c6686fb35e473dac2de6105eed62c6/src/lib/models/networks/dlav0.py,使用def dla34(pretrained, **kwargs): # DLA-34,在Tree block的self.project处failed

您好,请问解决了吗?

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

4 participants