我正在尝试向终端输出与 Keras 训练相同类型的训练进度条。我是 tensorflow 的新手,还没有尝试过 Keras,但我很想知道没有 Keras 是否可以完成。
请您参考如下方法:
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 原生打印函数。
