Jen*_*sun 5 python keras tensorflow pytorch
在编写机器学习模型时,我发现自己需要计算指标,或者在回调中运行额外的前向传递以实现可视化目的。在PyTorch中,我使用 执行此操作torch.no_grad(),这可以防止计算梯度,因此这些操作不会影响优化。
model(x)是可能的。但是,也可以说model.predict(x),这似乎也援引了call. 两者有区别吗?小智 6
张量流等效为tf.stop_gradient
另外不要忘记,Keras 在使用预测(或仅通过调用模型)时不会计算梯度__call__。
| 归档时间: |
|
| 查看次数: |
5106 次 |
| 最近记录: |