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

Pinning memory can cause errors for smaller GPUs #316

Open
jorshi opened this issue Nov 5, 2021 · 1 comment
Open

Pinning memory can cause errors for smaller GPUs #316

jorshi opened this issue Nov 5, 2021 · 1 comment

Comments

@jorshi
Copy link
Contributor

jorshi commented Nov 5, 2021

I'm getting a memory error trying to run predictions: RuntimeError: Caught RuntimeError in pin memory thread for device 0.

Setting pin_memory=False in the DataLoader in task_predictions.py solves this issue. https://github.com/neuralaudio/hear-eval-kit/blob/174b9268f96d3b4de51cd3557f9063d64fe1bd02/heareval/predictions/task_predictions.py#L744

Is this just for speed? Can we make this optional?

@turian
Copy link
Contributor

turian commented Nov 7, 2021

Yes, we can definitely make this option. We might consider making a config.py, separated into configs that affect results (this one doesn't) and ones that are basically just performance

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

No branches or pull requests

2 participants