This PR adds functionality for running the fft calculation of the gw loss with a pytorch model.
** note the linear algebra calculation approach has errors and should not be used with pytorch models until further debugging. (there is an assert line in the training code to enforce this)
This PR adds functionality for running the fft calculation of the gw loss with a pytorch model.
** note the linear algebra calculation approach has errors and should not be used with pytorch models until further debugging. (there is an assert line in the training code to enforce this)