首页 > 解决方案 > Tensorflow model.fit() 打印没有错误的重复消息

问题描述

我一直在尝试在 python 中使用 TensorFlow 在我的本地机器上训练一些神经网络(NN)。我得到这个奇怪的错误,model.fit 方法在单个时期的训练期间多次打印损失和其他指标。训练过程中没有错误,训练效果很好。只是重复的消息输出使我很难跟踪进度。

我尝试重新安装 tensorflow,但没有解决问题。我还训练了多个不同的 NN,有些是简单的 MLP,有些是 CNN。我所有的神经网络都存在这个问题。任何建议将不胜感激。

这是奇怪消息输出的屏幕截图: Weird message output

我想要这样的输出: 期望的输出

谢谢!

标签: pythontensorflow

解决方案


这真的很奇怪。在 model.fit 中,您没有指定详细参数。这里的文档没有定义默认模式,但显然它是 verbose=1 因为你得到了一个进度条。尝试设置 verbose=2 并查看打印是否正确。我假设您的输入是数组,对吗?如果是数组,数组的维度是多少?您没有指定 batch_size,因此默认为 32。打印输出显示每个 epoch 需要 586 步,这意味着您有大约 586 X 32=18752 个样本,对吗?


推荐阅读