google-deepmind / jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX
Apache License 2.0
87 stars 11 forks source link

Provide version numbers for Python packages #2

Closed Solosneros closed 2 years ago

Solosneros commented 2 years ago

Hi,

thanks for providing the source code to your "Unlocking High-Accuracy Differentially Private Image Classification through Scale" paper. I would be nice if you could provide version numbers for the required packages. So far, I have not been able to run your experiments. It seems that I run into errors with a lot of packages.

Thanks!

lberrada commented 2 years ago

In my setup, specifying jax==0.3.5 (like in this commit) within a fresh virtual environment solves the dependency issue.

Please let me know if that does not solve the problem on your end.

Solosneros commented 2 years ago

Thanks for the reply! I worked on it in the meantime and debugged the versions and can now at least execute the experiments with jax==0.3.5 up until another error that might be related to me not having access to TPUs. I will let you know.

lberrada commented 2 years ago

Great, resolving the issue since the dependency problem seems solved. Feel free to open a new issue if you encounter another error.