Skip to content

Conversation

xugaoxiang
Copy link

update for error:

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

@L1aoXingyu
Copy link
Member

Maybe already fix it in this commit(62ad5e1)

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.

2 participants