TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to

報錯如下所示 之後點擊最下面的一個.py文件,進入model_hook.py文件裏面,可以看到 直接修改源文件,在detach()後增加cpu()。 修改完成之後的代碼圖下所示 看,可以跑起來了,但是就是多了很多not supported! 這些東西怎麼刪除,我不知道了。等下研究
相關文章
相關標籤/搜索