我正在尝试训练一个 VGG16 模型,并且有 4k 个训练样本和 2k 个验证样本。
在上图中,尽管前 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/