如何在 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 中找到了這個解決方案。