随笔分类 -  深度学习error

RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
摘要：出错代码： 点击查看代码 dataset_numpy = x[0].cpu().numpy() centers = centers.cpu().numpy() codes = codes.cpu().numpy() 修改后代码： 点击查看代码 dataset_numpy = x[0].cpu().d 阅读全文
posted @ 2022-11-23 13:28 荒北 阅读(1355) 评论(0) 推荐(0) 编辑
Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select)
摘要：数据不在同一设备 将device = d2l.try_gpu() 改为device = torch.device('cpu') 62seq2seq 阅读全文
posted @ 2022-07-06 20:31 荒北 阅读(1314) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示