TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to numpy
D:\anaconda3\envs\PY3.8\Lib\site-packages\torch\tensor.py
在630行
return self.numpy()改成return self.cpu().numpy()
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to numpy
D:\anaconda3\envs\PY3.8\Lib\site-packages\torch\tensor.py
在630行
return self.numpy()改成return self.cpu().numpy()