python - Keras 训练进度条在一行上带有纪元号

Question

当我使用 Keras 通过 model.fit() 训练模型时，我看到一个进度条，如下所示:

Epoch 1/10
8000/8000 [==========] - 55s 7ms/step - loss: 0.9318 - acc: 0.0783 - val_loss: 0.8631 - val_acc: 0.1180
Epoch 2/10
8000/8000 [==========] - 55s 7ms/step - loss: 0.6587 - acc: 0.1334 - val_loss: 0.7052 - val_acc: 0.1477
Epoch 3/10
8000/8000 [==========] - 54s 7ms/step - loss: 0.5701 - acc: 0.1526 - val_loss: 0.6445 - val_acc: 0.1632

为了提高可读性，我希望将纪元编号与进度条放在同一行，如下所示:

Epoch 1/10: 8000/8000 [==========] - 55s 7ms/step - loss: 0.9318 - acc: 0.0783 - val_loss: 0.8631 - val_acc: 0.1180
Epoch 2/10: 8000/8000 [==========] - 55s 7ms/step - loss: 0.6587 - acc: 0.1334 - val_loss: 0.7052 - val_acc: 0.1477
Epoch 3/10: 8000/8000 [==========] - 54s 7ms/step - loss: 0.5701 - acc: 0.1526 - val_loss: 0.6445 - val_acc: 0.1632

我怎样才能做出改变？我知道 Keras 有可以在训练期间调用的回调，但我不熟悉它是如何工作的。

Answer 1

如果您想使用替代方案，可以使用 tqdm(版本 >= 4.41.0):

from tqdm.keras import TqdmCallback
...
model.fit(..., verbose=0, callbacks=[TqdmCallback(verbose=2)])

这会关闭 keras 的进度 (verbose=0)，并改用 tqdm。对于回调，verbose=2 表示 epochs 和 batches 的单独进度条。 1 表示完成后清除批处理栏。 0 表示只显示 epochs(从不显示 batch bars)。