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?
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 0After investigating the reason, I found that there is a convolution
self.projectafter 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?