trailofbits / PrivacyRaven

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

Separate model-specific and data-specific hyperparameters #50

Open suhacker1 opened 3 years ago

suhacker1 commented 3 years ago

Is your feature request related to a problem? Please describe. Right now, PrivacyRaven mixes all of the hyperparameters necessary to train a substitute model into a single dictionary. This should be replaced with Lightning Data Modules and/or other structures to clarify the role of each parameter and make it easier to extend PrivacyRaven for different tasks. This may require automatically turning all synthesized data into a Data Module.

Detail any additional context.