python - 如何在tensorflow中显示训练进度条？

Question

我正在尝试向终端输出与 Keras 训练相同类型的训练进度条。我是 tensorflow 的新手，还没有尝试过 Keras，但我很想知道没有 Keras 是否可以完成。

Answer 1

import tensorflow as tf
train_data = (...) 
progbar = tf.keras.utils.Progbar(len(train_data))

for i, d in enumerate(train_data):
    (train model here...)
    progbar.update(i) # This will update the progress bar graph.


3714/3715 [============================>.] - ETA: 20s

• 在 Tensorflow (v1.1~ 2.1) 中，建议使用 tf.keras.utils.Progbar() 而不是导入 tqdm。
• 仅供引用，使用 tf.print()，而不是 python 原生打印函数。