【问题标题】:What is the TensorFlow/Keras equivalent of PyTorch's `no_grad` function?PyTorch 的 `no_grad` 函数的 TensorFlow/Keras 等价物是什么?
【发布时间】:2021-05-04 13:38:53
【问题描述】:

在编写机器学习模型时,我发现自己需要计算指标,或在回调中运行额外的前向传递以实现可视化。在 PyTorch 中,我使用 torch.no_grad() 执行此操作,这样可以防止计算梯度,因此这些操作不会影响优化。

  1. 这种机制在 TensorFlow/Keras 中如何工作?
  2. Keras 模型是可调用的。所以,像model(x) 这样的东西是可能的。但是,也可以说model.predict(x),这似乎也调用了call。两者有区别吗?

【问题讨论】:

标签: python tensorflow keras pytorch


【解决方案1】:

等效的张量流是tf.stop_gradient

另外不要忘记,Keras 在使用预测时不会计算梯度(或者只是通过 __call__ 调用模型)。

【讨论】:

  • 通过__call__ 调用和使用.predict 调用有区别吗?
猜你喜欢
  • 2022-06-22
  • 2019-11-11
  • 2020-12-28
  • 1970-01-01
  • 2019-08-29
  • 1970-01-01
  • 2017-01-14
  • 2022-08-21
  • 2019-03-24
相关资源
最近更新 更多