Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix num_cpus flag #150

Merged
merged 2 commits into from
Nov 21, 2022
Merged

fix num_cpus flag #150

merged 2 commits into from
Nov 21, 2022

Conversation

jpata
Copy link
Owner

@jpata jpata commented Nov 21, 2022

Fix the num_cpus flag to actually tell tensorflow to use a fixed number of threads. Useful to limit the application to a fixed CPU budget on a HPC.

In the previous version, using MirroredStrategy on CPU just blows up the RAM by a factor of Nthreads, which does not seem correct.

@jpata jpata merged commit 2d0ea76 into main Nov 21, 2022
@jpata jpata deleted the fix_num_cpus branch December 22, 2022 14:09
jpata added a commit that referenced this pull request Sep 15, 2023
* fix the num_cpus flag to actually change the number of cpus

* fix eval, clic postprocessing
jpata added a commit that referenced this pull request Sep 15, 2023
* fix the num_cpus flag to actually change the number of cpus

* fix eval, clic postprocessing

Former-commit-id: cf35d66
jpata added a commit that referenced this pull request Sep 15, 2023
* fix the num_cpus flag to actually change the number of cpus

* fix eval, clic postprocessing

Former-commit-id: cf35d66
jpata added a commit that referenced this pull request Sep 25, 2023
* fix the num_cpus flag to actually change the number of cpus

* fix eval, clic postprocessing
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant