如何在 tf.keras 训练期间获得当前时代的进度？

Question

我正在编写一个 GUI 应用程序来训练各种基于tf.keras的模型。 所以所有关于accuracy 、 loss等的信息都应该在 GUI 上可见，而不是默认的 Keras 控制台 output。

现在，我设法通过自定义tf.keras.callbacks.Callback class 在 model 训练期间获取所有相关信息，效果很好。 但我也想获得当前纪元的进度，即到目前为止已经训练了当前纪元的百分比，即 Keras 在训练期间通过进度条打印到控制台的内容。

有没有办法在 Keras 回调中检索该信息？

Answer 1

好的，找到了解决方案。 它可以在Callback class 的self.params属性中找到。
对我来说，这有效：

def on_epoch_begin(self, epoch, logs=None):
    self.epoch_step = 0

def on_batch_end(self, batch, logs=None):
    self.epoch_step += 1
    progress = self.epoch_step / self.params["steps"]

对于 reords，我在keras-tqdm GitHub repo 中找到了这个解决方案。