Viewing a single comment thread. View all comments

shahaff32 OP t1_is4jcv4 wrote

Reply to comment by danny_fel in [R] Wavelet Feature Maps Compression for Image-to-Image CNNs by shahaff32

Thanks :)

In the current state the implementation is using only standard Pytorch operations, therefore it is not as optimal as it can be, and the overhead of the wavelet transforms can outweighs the speedup of the convolution.

We are currently working on a CUDA implementation to overcome that :) see Appendix H for more details

1

danny_fel t1_is8eaky wrote

Oh thanks! Probably will play around with it! This sounds exciting from a maker/hobbyist perspective wanting to do edge applications.

2