setting device to lanzcos_low_rank_hessian_approx

aai-institute / pyDVL

pyDVL is a library of stable implementations of algorithms for data valuation and influence function computation

https://pydvl.org

GNU Lesser General Public License v3.0

100 stars 8 forks source link

Closed Xuzzo closed 1 year ago

Xuzzo commented 1 year ago

This PR closes #391

adds a warning for low rank approximation when device is gpu
re allocates eigenvals and eigenvecs to gpu before returning in lanzcos_low_rank_hessian_approx

[ ] Wrote Unit tests (if necessary)
[ ] Updated Documentation (if necessary)
[ ] Updated Changelog
[ ] If notebooks were added/changed, added boilerplate cells are tagged with "nbsphinx":"hidden"

schroedk commented 1 year ago

@Xuzzo Most probably, there will be conflicts with #394 . I will resolve them, depending on which branch is merged first.