Keras:为什么整个时代需要更长时间才能显示所有批次都已完成?

时间:2017-04-22 17:38:37

标签: 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)

1 个答案:

答案 0 :(得分:7)

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