yolact-pytorch icon indicating copy to clipboard operation
yolact-pytorch copied to clipboard
verified publisher icon bubbliiiing

不能转换CUDA tensor格式的数据，可以指导一下怎么怎么解决嘛

Open yums24 opened this issue 4 years ago • 1 comments

报错：TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

image

yums24 avatar Feb 18 '22 11:02 yums24

报错都让你先.cpu()

bubbliiiing avatar Feb 25 '22 03:02 bubbliiiing