search

privacytrustlab / ml_privacy_meter

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

OOM problem in attack alexnet #32

Closed detaimee closed 1 year ago

detaimee commented 3 years ago

Hi. When I run the alexnet.py code to create the model and then run the attack_alexnet.py code, I have an oom problem, so I ask a question. First of all, my setting environment was run with rtx 2080ti and I used tensorflow-gpu version 2.1.0 as in requirements.txt. So what I want to ask is

  1. What setting environment was used to run the code, and I am curious about the minimum graphic specification or graphic memory to execute the code.
  2. I am curious to know which variables need to be changed in order to be executed properly with the current graphics card.

Thanks in advance.

스크린샷 2021-03-17 오후 8 01 16
amad-person commented 3 years ago

Hi @detaimee, you can try to reduce the batch sizes for training/testing datasets. I also found this stackoverflow question that has some other approaches you can try.