我正在尝试训练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分钟。