pytorch_wavelets icon indicating copy to clipboard operation
pytorch_wavelets copied to clipboard
verified publisher icon fbcotter

Support Automatic Mixed Precision (AMP)

Open Queuecumber opened this issue 2 years ago • 2 comments

This PR adds the necessary decorators to the custom autograd functions to support AMP.

DTCWT forward and backward were tested on a few examples, other functions may require testing

Fixes #45

Queuecumber avatar Aug 02 '23 14:08 Queuecumber

@fbcotter Could you take a look at this please?

Queuecumber avatar May 09 '24 18:05 Queuecumber