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

Add `allow_skip` to `finalize_scores` #19

Closed kristian-georgiev closed 1 year ago

kristian-georgiev commented 1 year ago

Allow gathering scores even when some of the grads_target computations failed.

kristian-georgiev commented 1 year ago

https://github.com/MadryLab/trak/commit/2a6fec61850796ac4c8c1208f9c586945d848986