tensorflow / tensorflow/privacy
mnist_dpsgd_tutorial is very slow
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 2k
- Forks
- 477
- Avg merge
- 22h 12m
- Merged PRs (30d)
- 1
Description
I am following the blog regarding the TF-Privacy which basically follows code from two files mnist_scratch.py and mnist_dpsgd_tutorial.py to give the overview.
While mnist_scratch.py is running fine, the other file with the given default setting for flag 'dpsgd' as True is running very slow in training.
It is taking around 990 secs for 1 epoch when DP is enabled and 12 secs when it is disabled.
Could you please help in checking the issue?
I have seen a similar kind of issues but could not find the resolution.
Thanks
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by comparing mnist_scratch.py with mnist_dpsgd_tutorial.py, running the tutorial with the dpsgd flag enabled and disabled. Investigate the reported roughly 990-second versus 12-second epoch times and identify the cause of the slowdown. Done means the performance issue is explained and a corrective change or clear resolution is provided.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, tensorflow
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100