python - Keras 中是否有与 Caffe 中的 iter_size 等效？

Question

我一直在 Caffe 但在 Keras 中寻找等效的 iter_size 参数。

这就是 iter_size 的意思：

iter_size：通过 iter_size 求解器字段累积跨批次的梯度。使用此设置 batch_size: 16 和 iter_size: 1 和 batch_size: 4 和 iter_size: 4 是等价的。

我已经检查了 Keras 中的 steps_per_execution 参数，但我不确定它是否相同。

steps_per_execution：默认为 1。每次 tf.function 调用期间要运行的批处理数。在单个 tf.function 调用中运行多个批处理可以极大地提高 TPU 或具有大量 Python 开销的小型模型的性能。每次执行最多将运行一个完整的 epoch。如果传递的数字大于 epoch 的大小，则执行将被截断为 epoch 的大小。请注意，如果steps_per_execution 设置为N，Callback.on_batch_begin 和Callback.on_batch_end 方法将仅每N 个批次调用一次（即在每次tf.function 执行之前/之后）

有谁知道这是否相同或是否有另一个等效参数？我真的很感谢你的帮助，谢谢！

标签： pythonkerastensorflow2.0caffe