search

privacytrustlab / ml_privacy_meter

Privacy Meter: An open-source library to audit data privacy in statistical and machine learning algorithms.
MIT License
556 stars 99 forks source link

Better support for GPU #91

Closed changhongyan123 closed 1 year ago

changhongyan123 commented 1 year ago

In this PR, we update the PytorchTensor model in order to reduce GPU memory consumption. In particular, we move the model to GPU only when computing the signals.