search

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

Fix model ID loading #8

Closed bencw99 closed 1 year ago

bencw99 commented 1 year ago

Model IDs are converted to strings during serialization (because of json). This converts them back to integers during loading.