A fast, effective data attribution method for neural networks in PyTorch
Bug fixes and enhancements:
trak/contrib
(see CONTRIBUTING.md)0.3.0 by @kristian-georgiev and @AlaaKhaddaj in https://github.com/MadryLab/trak/pull/50
ChunkedCudaProjector
, (much) faster scoring by removing I/O bottlenecks)0.2.2 by @kristian-georgiev in https://github.com/MadryLab/trak/pull/49
more tests
better formatting
minor bug fixes:
controllable random seed for projector
fix bug with dtype and device of gradients
fix bug with init_projector when device is CPU
Co-authored-by: Sung Min Park spark@mslurm Co-authored-by: Alaa Khaddaj [email protected]
Full Changelog: https://github.com/MadryLab/trak/compare/v0.2.1...v0.2.2
bug fixes
updated docs
Full Changelog: https://github.com/MadryLab/trak/compare/v0.2.0...v0.2.1
Some (mild) backward incompatibilities introduced. In particular,
exp_name
is now a required argument when scoring.
handle pre-emption for featurizing
support scoring & featurizing data shards in parallel
reduce memory footprint by ~1.5x
migrate to torch.func
bump torch dep requirement to 2.0.0 because of torch.func
python >=3.8 for pytorch 2.0
project and store in float16 by default
tie experiment name to scoring targets; simplify saver; add logging
save scores as mmap
normalization factor for numerical stability
clean up quickstart
no-op projector
pass in an instance of a class for tasks, rather than init inside of gradientcomputer
bug fixes
@AlaaKhaddaj made their first contribution in #38
Full Changelog: https://github.com/MadryLab/trak/compare/v0.1.3...v0.2.0
0.1.3 by @kristian-georgiev in https://github.com/MadryLab/trak/pull/32
allow skipping model IDs in finalize scores
allow subclassing of saver and score_computer directly from traker args
default to BasicProjector if CudaProjector projeciton step errors out
add another type of error that sometime occurs when fast_jl has issues
update quickstart notebook
Add link to colab with pre-computed trak scores to readme
add dropbox links to quickstart nb
update training code in quickstart tutorial
bump version
Full Changelog: https://github.com/MadryLab/trak/compare/v0.1.2...v0.1.3
Full Changelog: https://github.com/MadryLab/trak/compare/v0.1.1...v0.1.2
trak v0.1.1
Full Changelog: https://github.com/MadryLab/trak/compare/v0.1.0...v0.1.1
trak v0.1.0