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

问题描述 投票:2回答:1

我正在尝试训练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)

tensorflow deep-learning keras
1个回答
7
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.