在pytorch中,直接从cuda取数据,需要加上.cpu()
改为imgs_hr.cpu().numpy()
TypeError: can‘t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor
最新推荐文章于 2023-01-03 20:03:38 发布
在pytorch中,直接从cuda取数据,需要加上.cpu()
改为imgs_hr.cpu().numpy()