如何在回调中获得批次中使用的确切训练示例？

Question

我在 Keras 中训练神经网络时遇到问题。每个 epoch，损失都会稳步下降，达到大约 1e-9，然后在 epoch 中间的某个地方（可能是任何地方），损失上升到 5e-5，并最终稳定在每个 epoch 相同的最终损失上。我相信这是由于我的数据集中的一些脏数据导致模型无法训练超过某个点，尽管我真的不确定。

为了测试我的假设，我想创建一个自定义的 Keras 回调对象，该对象将确定批次后的损失是否有足够大的跳跃，并指出哪个批次导致了跳跃。问题是batch提供给的参数keras.callbacks.Callback.on_batch_end只是批次编号，而不是该批次中使用的实际训练示例。此外，logs传入的dict 也仅包含loss和acc。

这意味着我实际上无法确定哪些数据导致了损失的增加。有没有办法可以确定导致每个时期跳跃的确切训练示例？有什么方法可以在回调中访问它吗？