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.

Sign up for GitHub

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

Jump to bottom

Fix JAX resource exhaustion/multi-threading #16

Open

1 of 2 tasks

brianreicher opened this issue Aug 15, 2022 · 0 comments

Open

1 of 2 tasks

Fix JAX resource exhaustion/multi-threading #16

brianreicher opened this issue Aug 15, 2022 · 0 comments

Assignees

Labels

bug

Collaborator

brianreicher commented Aug 15, 2022 •

edited

Loading

JAX network testing running into two main issues, look to collapse one at a time:

Single GPU training is able to load the untrained network onto the GPU but unable to allocate resources to run convolutions
Multi-threaded training splits the first input dimension (batch size=16) into separate (4,4) dimensions, causing the rank of the input tensors to be expanded

brianreicher added the bug label

brianreicher self-assigned this

brianreicher added a commit that referenced this issue


          Update jax test to point of rank issue #14 #16

d37b79f

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment