pytorch

Question

pytorch

GF1447571253 opened this issue 3 years ago · 2 comments

GF1447571253 commented 3 years ago

can adapool function completely implement in torch，without any cpp

GF1447571253 commented 3 years ago

ok,thanks

Answer 1 · 2021-11-29T15:10:48.000Z

Hi @GF1447571253,

There are no plans currently to have a pure pytorch implementation. Mainly because of the computational and memory requirements (since the intention is to have a lightweight and usable pooling method). You could however try to implement it yourself based on the paper ( you can draw inspiration from this post for DSC with the target, in this case, being avg_pool.