Support GPU computation

open-forest-observatory / tree-detection-framework

BSD 3-Clause "New" or "Revised" License

3 stars 0 forks source link

Support GPU computation #45

Closed russelldj closed 3 weeks ago

russelldj commented 3 weeks ago

This moves the data to the correct device in the forward method of the model and sets the device for the model to GPU if present in the generate_predictions function.