参数记得加requires_grad=True
loss = torch.rand(24,requires_grad=True)
就好了
element 0 of variables does not require grad and does not have a grad_fn
最新推荐文章于 2023-08-07 09:36:49 发布
在PyTorch中,设置requires_grad=True是开启梯度计算的关键,这在训练神经网络时尤其重要。loss=torch.rand(24,requires_grad=True)这行代码创建了一个随机张量并启用了梯度追踪,以便于后续反向传播计算损失的梯度。
摘要由CSDN通过智能技术生成