2019年4月9日
TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
摘要： 报错原因：numpy不能读取CUDA tensor 需要将它转化为 CPU tensor。 所以如果想把CUDA tensor格式的数据改成numpy时，需要先将其转换成cpu float-tensor随后再转到numpy格式 报错行： 修改后： 阅读全文
posted @ 2019-04-09 15:46 小呆丶 阅读(12972) 评论(0) 推荐(1) 编辑