Skip to content

added cpu support for test.py#168

Open
nickren1234 wants to merge 1 commit into
NVIDIA:masterfrom
nickren1234:cpu-support
Open

added cpu support for test.py#168
nickren1234 wants to merge 1 commit into
NVIDIA:masterfrom
nickren1234:cpu-support

Conversation

@nickren1234
Copy link
Copy Markdown

input tensor was being pushed to cuda even if the cpu only flag is on.

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.

1 participant