nni icon indicating copy to clipboard operation
nni copied to clipboard

[New Feature Request] support 3D model compression

Open longfei98 opened this issue 4 years ago • 3 comments

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.

longfei98 avatar Feb 04 '21 08:02 longfei98

You are right, we do not support Conv3d yet.

zheng-ningxin avatar Feb 05 '21 09:02 zheng-ningxin

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.

longfei98 avatar Feb 06 '21 13:02 longfei98

Any plan for supporting Conv3D?

mjlsuccess avatar May 19 '22 01:05 mjlsuccess