search

v0lta / PyTorch-Wavelet-Toolbox

Differentiable fast wavelet transforms in PyTorch with GPU support.
https://pytorch-wavelet-toolbox.readthedocs.io
European Union Public License 1.2
279 stars 36 forks source link

Incorporate learnable wavelet layer #80

Closed cthoyt closed 8 months ago

cthoyt commented 8 months ago

Closes #79

v0lta commented 8 months ago

I am not convinced the layer we proposed back in 2020 is still relevant for current architectures ( see also https://github.com/v0lta/PyTorch-Wavelet-Toolbox/issues/79 ). The idea of having the 'nn' namespace is nice, but we would need something people can just plug in with good performance, the compressed layer will most likely not do that.