深度学习遇到的问题③: Can‘t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.

问题描述:

在模型训练过程的可视化中:

 

运行出现以下错误:Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.

 

 问题分析:

 张量Train_loss_all本身带有梯度,

直接将其转换为numpy数据将破坏计算图,因此numpy拒绝进行数据转换.、故在变量转换之前添加detach()调用。

 解决办法:

将张量Train_loss_all,Test_loss_all自行转化为numpy数组。代码:
train_loss_all_np=[]
test_loss_all_np=[]
for i in range(30):
    train_loss_all_np.append(train_loss_all[i].detach().numpy())
    test_loss_all_np.append(test_loss_all[i].detach().numpy())

并将标注那两行修改为
plt.plot(train_loss_all_np,"ro-",label="Train loss")
plt.plot(test_loss_all_np,"bs-",label="Test loss")

即可实现模型训练过程的可视化

 

物联沃分享整理
物联沃-IOTWORD物联网 » 深度学习遇到的问题③: Can‘t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.

发表评论