时间戳

首页 > 解决方案 > 如何在tensorflow中显示训练进度条？

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

问题描述

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

标签： pythontensorflow

解决方案

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 本机打印功能。

推荐阅读