search

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
89 stars 9 forks source link

Missing move to device in Lissa implementation #494

Closed schroedk closed 5 months ago

schroedk commented 5 months ago

In the implementation of the Lissa algorithm, the tensors from the training dataloader are not moved to the torch model's device. To execute on any device other than CPU, one needs to move the dataset manually to the device.