Pytorch backward报错2次访问计算图需要 retain_graph=True 的一种情况

最新推荐文章于 2024-04-28 20:57:20 发布

培之

最新推荐文章于 2024-04-28 20:57:20 发布

阅读量796

点赞数 10

CC 4.0 BY-SA版权

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/OrdinaryMatthew/article/details/136172926

错误代码

错误的原因在于

y1 = 0.5*x*2-1.2*x
y2 = x**3

没有放到循环里面，没有随着 x 的优化而相应变化。

import torch
import numpy as np
import torch.optim as optim

torch.autograd.set_detect_anomaly(True)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
x = torch.tensor([1.0, 2.0, 3.0,4.5], dtype=torch.float32, requires_grad=True, device=device)


y_GT= torch