Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

do not pin memory when device is CPU to avoid warning #1400

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

Open
pedrormjunior wants to merge 1 commit into JaidedAI:master
base: master
Choose a base branch
Loading
from pedrormjunior:fix_pin_memory_warning

Conversation

Copy link

@pedrormjunior pedrormjunior commented Apr 27, 2025

This seems to fix issue #854.

Without this modification, it raises the following warning:

/path/to/.venv/lib/python3.13/site-packages/torch/utils/data/dataloader.py:665: UserWarning: 'pin_memory' argument is set as true but no accelerator is found, then device pinned memory won't be used.
 warnings.warn(warn_msg)

I did not think about collateral effects of this change, other use cases, etc. Please, check.

zperk13 reacted with thumbs up emoji
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

AltStyle によって変換されたページ (->オリジナル) /