fbcotter / pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet
Other
943 stars 146 forks source link

How to calculate the params of DWT and IDWT? #54

Open lijun2005 opened 9 months ago

lijun2005 commented 9 months ago

I think the trainable params of DWT and IDWT should be 0, but I have tried thop and torchinfo two libraries which indicate params is not 0.