MadryLab / trak

A fast, effective data attribution method for neural networks in PyTorch
https://trak.csail.mit.edu/
MIT License
180 stars 26 forks source link

Do not allocate train x target score matrix for each model ID #13

Closed kristian-georgiev closed 1 year ago

kristian-georgiev commented 1 year ago

https://github.com/MadryLab/trak/commit/dab584d4d5db713d206afa31a323f0dfa179610b