search

BigDataWUR / AgML-CY-Bench

CY-Bench (Crop Yield Benchmark) is a comprehensive dataset and benchmark to forecast crop yields at subnational level. CY-Bench standardizes selection, processing and spatio-temporal harmonization of public subnational yield statistics with relevant predictors. Contributors include agronomers, climate scientists and machine learning researchers.
https://cybench.agml.org/
Other
9 stars 3 forks source link

Fix for use with CUDA #236

Closed hbja closed 3 weeks ago

hbja commented 3 weeks ago

Propagated input in forward() to GPU if using cuda. Otherwise it complains inputs and model are in different devices.

krsnapaudel commented 3 weeks ago

@hbja Please run black before merging. python -m black . from the repo root.