search

MadryLab / trak

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

fix: CLIP out to loss grad dim #72

Closed dwahdany closed 2 months ago

dwahdany commented 2 months ago

closes #73

kristian-georgiev commented 2 months ago

Thank you!