获取对Tensorflow/Keras回调中的历元和批次总数的访问

Question

对于一个定制的监控工具，我试图使用历元和批次的总数来计算完成的估计时间，并显示还有多少历元/批次需要完成。我正在使用PyTorch Lightning和Tensorflow回调。在PyTorch Lightning中，我可以通过trainer.max_epochs、trainer.num_training_batches和trainer.num_val_batches轻松访问所需的值

然而，我仍在寻找Tensorflow回调中的对应项self.model与PyTorch的trainer没有类似的属性。尽管TensorBoard和TQM等工具可以显示进度条，但必须有一种方法来访问回调中的历元/批总数，而不必显式地传递它们。但是我还没有找到任何有用的东西