search

bubbliiiing / yolact-pytorch

这是一个yolact-pytorch的库，可用于训练自己的数据集
MIT License
70 stars 12 forks source link

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

Open yums24 opened 2 years ago

yums24 commented 2 years ago

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

image

bubbliiiing commented 2 years ago

报错都让你先.cpu()