python - 咖啡乐网 : Difference between `solver.step(1)` and `solver.net.forward()`

Question

我正在查看 Caffe LeNet 教程 here我想到了一个问题:

这两个代码有什么区别:

self.solver.step(1)

和

self.solver.net.forward()  # train net

至少根据评论，他们似乎都在训练网络。

我个人认为第一个在训练数据上训练网络并更新 net 和 test_net 的权重，但第二个似乎只转发了一批数据并应用从上一步学到的权重。

如果我认为是对的，那么教程中第二段代码的目的是什么？为什么代码执行 net.forward ？ solver.step(1) 不能自己做吗？

谢谢你的时间

Answer 1

step 进行一次完整迭代，涵盖所有三个阶段:forward 评估、backward 传播和更新 .对 forward 的调用只执行其中的第一个。签名(参数列表)也有差异。