tensorflow - Keras:为什么显示所有批处理都已完成时,整个 epoch 需要更长的时间?

标签 tensorflow deep-learning keras

我正在尝试训练一个 VGG16 模型,并且有 4k 个训练样本和 2k 个验证样本。

enter image description here

在上图中,尽管前 134 批完成得非常快,但最后一批等待很长时间,大约 8 分钟后完成,我认为这花费了太长时间。我做错了什么吗?使用以下内容开始训练过程。

model.compile(optimizer='rmsprop', loss='categorical_crossentropy',metrics=['accuracy'])

model.fit_generator(train_generator, samples_per_epoch=4320, epochs=50, validation_data=validation_generator, nb_val_samples=2880)

最佳答案

当一个时期结束时,当您提供验证数据时,Keras 必须在评估集上评估您的模型,这在您的情况下需要 8 分钟。

关于tensorflow - Keras:为什么显示所有批处理都已完成时,整个 epoch 需要更长的时间?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43562426/

相关文章:

python - Tensorflow:Py_func 返回未知形状

python - TensorFlow - 导入元图并使用其中的变量

machine-learning - LSTM 神经网络的错误预测

python - 喀拉斯 : training with an array as an input

python-3.x - 如何修复当我尝试在终端上运行构建图时弹出的张量板错误

tensorflow - CNN(Conv Neural Net)如何避免GPU的瓶颈?

python - 乘法 tf.range() 的优雅方式

machine-learning - 检查失败 : mdb_status == 0 (2 vs. 0) 没有这样的文件或目录

python - 添加 LSTM 层但获得所需的位置参数 : 'units' error

keras - 我如何在 keras 中使用带有灰度图像的神经网络