Keep portion update - Githubissues

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

Closed SimonTopp closed 2 years ago

SimonTopp commented 2 years ago

Caught a bug in the keep portion conditional statements in torch_utils. Fixed in order to be able to reshape RGCN data for GWN.
Fixed RGCN keep portion conditional in train routine.
Changed to infer device for predicting torch models rather than passing an argument.