【发布时间】:2021-05-04 13:38:53
【问题描述】:
在编写机器学习模型时,我发现自己需要计算指标,或在回调中运行额外的前向传递以实现可视化。在 PyTorch 中,我使用 torch.no_grad() 执行此操作,这样可以防止计算梯度,因此这些操作不会影响优化。
- 这种机制在 TensorFlow/Keras 中如何工作?
-
Keras 模型是可调用的。所以,像
model(x)这样的东西是可能的。但是,也可以说model.predict(x),这似乎也调用了call。两者有区别吗?
【问题讨论】:
标签: python tensorflow keras pytorch