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

Question

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

Closed this issue 2 years ago · 1 comments

model.py的第40行“ item_embeddings = np.sum(final, 0) / (self.layers+1)” 报错：
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

Process finished with exit code 1

Answer 1 · 2022-12-15T13:20:59.000Z

please refer to #9 (comment)