search

USGS-R / river-dl

Deep learning model for predicting environmental variables on river systems
Creative Commons Zero v1.0 Universal
21 stars 15 forks source link

adding pytorch implementation of gw loss #171

Closed janetrbarclay closed 2 years ago

janetrbarclay commented 2 years ago

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)