Closed SimonTopp closed 3 years ago
just adding a note to this conversation that Simon and I are working on debugging this. He can get it to work locally and on TG. I am getting errors on both pangeo and TG.
seems to be a cpu / gpu issue. the fft approach fails on the gpu's and runs correctly on cpu's
Updated GW loss to use the FFT approach. One important distinction is that we now calculate the air phase and amplitude yearly within the loss function instead of beforehand in the data prep.