search

MadryLab / trak

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

Migrate from `functorch` to `torch.func` #34

Closed kristian-georgiev closed 1 year ago

kristian-georgiev commented 1 year ago

This will change the traker requirements from torch>=1.13 to torch>=2.0.0

kristian-georgiev commented 1 year ago

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

kristian-georgiev commented 1 year ago

https://github.com/MadryLab/trak/commit/8191f3b3933da1be31b0b99aeea94eb85693d105