【问题标题】:tensorflow stop_gradient equivalent in pytorchpytorch 中的 tensorflow stop_gradient 等效项
【发布时间】:2019-01-02 22:14:35
【问题描述】:

什么是 pytorch 中的 tf.stop_gradient() 等效项(提供一种在反向传播期间不计算某些变量的梯度的方法)?

【问题讨论】:

标签: tensorflow pytorch


【解决方案1】:

pytorch 中的张量具有requires_grad 属性。将其设置为 False 以防止该张量的梯度计算。

【讨论】:

    【解决方案2】:

    你能用 x.detach() 检查一下吗?

    猜你喜欢
    • 2021-11-29
    • 2022-10-12
    • 2021-07-04
    • 2020-10-30
    • 2021-08-01
    • 2020-03-28
    • 2021-01-31
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多