【发布时间】:2019-01-02 22:14:35
【问题描述】:
什么是 pytorch 中的 tf.stop_gradient() 等效项(提供一种在反向传播期间不计算某些变量的梯度的方法)?
【问题讨论】:
标签: tensorflow pytorch
什么是 pytorch 中的 tf.stop_gradient() 等效项(提供一种在反向传播期间不计算某些变量的梯度的方法)?
【问题讨论】:
标签: tensorflow pytorch
pytorch 中的张量具有requires_grad 属性。将其设置为 False 以防止该张量的梯度计算。
【讨论】:
你能用 x.detach() 检查一下吗?