更改 Keras train_on_batch() 的详细程度？

Question

我正在使用 Keras 的

train_on_batch()

命令训练 GAN。这与 Keras 的

fit()

非常相似。但是，在

fit()

的文档中，有一个

verbose

的参数，它可以更改将进度条打印到控制台的频率。

我的模型有很多批次，因此它会向命令行打印大量进度条。不幸的是，

train_on_batch()

没有

verbose

参数。有解决方法吗？是否有我可以设置的 Keras 全局变量/环境变量？我不想禁止我的程序打印到控制台，我只想更改具体的详细程度

train_on_batch()

。

为了澄清，我直接从 Keras 包使用 Keras，而不是使用 tf.keras。

Answer 1

您可以通过调用以下方法在训练循环之前禁用所有进度条：

tf.keras.utils.disable_interactive_logging()

训练循环结束后，通过调用启用方法来启用进度条：

tf.keras.utils.enable_interactive_logging()