search

conservationtechlab / animl-py

Animl comprises a variety of machine learning tools for analyzing ecological data. This Python package includes a set of functions to classify subjects within camera trap field data and can handle both images and videos.
MIT License
6 stars 3 forks source link

Run Onnx on GPU #58

Closed tkswanson closed 2 weeks ago

tkswanson commented 1 month ago

Use IOBinding to access torch tensors already on gpu from data generator instead of sending to cpu