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

Question

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

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

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

Answer 1

好，找到解决方案。可以在self.params类的Callback属性中找到。
对我来说，这可行：

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"]

对于老兄来说，我找到了这个解决方案here in the keras-tqdm GitHub repo。