The reason for this error is that numpy is used to calculate the tensor in the previous cuda. Check the code and use A.data.cpu().numpy() to transform it to continue
[Linux server error] TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor
Guess you like
Origin blog.csdn.net/qq_45347185/article/details/108965347
Ranking