trailofbits / PrivacyRaven

Privacy Testing for Deep Learning
Apache License 2.0
184 stars 16 forks source link

Create a wrapper around PyTorch Lightning Callbacks #37

Closed suhacker1 closed 3 years ago

suhacker1 commented 4 years ago

Is your feature request related to a problem? Please describe. PyTorch Lightning callbacks enable users to extend the functionality after already creating the attack. In other words, this should be a function that takes in the attack output itself.

Detail any additional context. If there are any callbacks that should be integrated within PrivacyRaven, create an issue or comment on this one.