-
Notifications
You must be signed in to change notification settings - Fork 338
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
Comments
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. |
谢谢大佬的回复,这个链接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) |
您好,请问解决了吗? |
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?
The text was updated successfully, but these errors were encountered: