python - train_on_batch() 在 keras 模型中做了什么？

Question

我看到了一个代码示例(太大而无法粘贴到此处)，其中作者使用了 model.train_on_batch(in, out) 而不是 model.fit(in, out)。 Keras 的官方文档说:

Single gradient update over one batch of samples.

但是我不明白。它是否与 fit() 相同，但不是执行许多前馈和反向传播步骤，而是执行一次？还是我错了？

Answer 1

是的，train_on_batch 仅使用单个批处理进行一次训练。

虽然 fit 为许多时期训练了许多批处理。 (每个批处理都会导致权重更新)。

使用 train_on_batch 的想法可能是在每个批处理之间自己做更多的事情。