search

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

Compatiable with torch AMP? #45

Open abeyang00 opened 2 years ago

abeyang00 commented 2 years ago

It seems like this module is not trained with mixed precision.

Do you plan on adding this feature?

Thank you,