Skip to content
This repository has been archived by the owner on Dec 2, 2024. It is now read-only.

return x.cuda(async=True) if torch.cuda.is_available() else x #35

Open
aadoranti-vs opened this issue Jun 1, 2022 · 0 comments
Open

Comments

@aadoranti-vs
Copy link

async is now a keyword. It should be changed to 'non_blocking'

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant