TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
具体报错如下图所示:
此时打开utils.py文件,修改为下图所示代码即可
成功!!

TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
具体报错如下图所示:
此时打开utils.py文件,修改为下图所示代码即可
成功!!