VainF / Torch-Pruning

[CVPR 2023] DepGraph: Towards Any Structural Pruning
https://arxiv.org/abs/2301.12900
MIT License
2.68k stars 331 forks source link

deformable convolution #413

Closed AvrahamRaviv closed 2 months ago

AvrahamRaviv commented 2 months ago

Hi Does your code support deformable convolution (deform_conv2d)? Thanks

janthmueller commented 2 months ago

It doesn't provide one out of the box, but you can define an appropriate one for it as in function.py or as explained in the wiki for customized layers.

AvrahamRaviv commented 2 months ago

Thanks!