nni
nni copied to clipboard
[New Feature Request] support 3D model compression
I checked the doc and some codes to find that many pruners only suport conv2d. Is there any way to prune 3D model?
It seems that only fine-grained pruning (Like Level?) do not be limited by 2d while it can not be speed up by nni. Because nni only support speed up coarse-grained pruning in current version, is my understanding correct?
Thank you very much.
You are right, we do not support Conv3d yet.
You are right, we do not support Conv3d yet. Thank you for your reply. Is there any plan in future? Before that, I can modify some specified pruning method by myself. While I am still looking forward to use it in nni. It is really a nice job.
Any plan for supporting Conv3D?