Skip to main content
 首页 » 编程设计

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

2024年08月13日36mayingbao

我正在尝试向终端输出与 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 原生打印函数。