Closed TheaperDeng closed 3 months ago
will take a look later
will take a look later
As mentioned in #85 , probably we can test TRAK performance on CIFAR-2 also? will need to add one more example script.
Hi @TheaperDeng , overall, do you think the unsqueeze(0)
style stuff within the f,m
function is acceptable?
Not sure whether we can get rid of that by some different vmap
configuration.
Hi @TheaperDeng , overall, do you think the
unsqueeze(0)
style stuff within thef,m
function is acceptable? Not sure whether we can get rid of that by some differentvmap
configuration.
now leave this as a TO-DO, please refer to #86 .
Description
1. Motivation and Context
Add TRAK as one of the attributor in our package
2. Summary of the change
TRAKAttributor
with similar usage as other attributor. The only difference is that it requires acorrect_possibility_func
for the calculation of Q.3. What tests have been added/updated for the change?