Pruning: Export ONNX with pruned connections removed #588
Unanswered
saravanabalagi
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Say I do filter pruning, does mmrazor change the architecture or retain a prune mask which gets applied when model forward is called similar to the one in torch.nn.utils.prune?
When I export ONNX will the pruned connections be removed completely? i.e., I am not looking for weights filled with zeros making it sparse, instead I want the filters/channels/neurons to be removed completely from the network. Is this possible using mmrazor?
Beta Was this translation helpful? Give feedback.
All reactions