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

Use multi gpu for conditional moment estimation #157

Merged
merged 3 commits into from
Jul 19, 2023

Conversation

malihass
Copy link
Collaborator

@malihass malihass commented Jul 18, 2023

closes #156

  • Mask needed to go through split_array if used to compute the loss
  • Also update documentation for setting up training on GPU. The installation of tensorflow-gpu needs to be done after sup3r install to avoid overriding TensorFlow version. Some extra library need to be installed on Eagle as well.

@malihass malihass requested a review from bnb32 July 18, 2023 22:40
@malihass malihass merged commit 9c097f1 into main Jul 19, 2023
5 checks passed
@malihass malihass deleted the malihass/multiGPU_cond_mom branch July 19, 2023 19:04
github-actions bot pushed a commit that referenced this pull request Jul 19, 2023
Use multi gpu for conditional moment estimation
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.

Multi gpu for conditional moment
2 participants