Keras：爲什麼整個時期顯示所有批次都完成時需要更長的時間？

我想訓練一個VGG16模型，並有4k訓練樣本和2k驗證樣本。Keras：爲什麼整個時期顯示所有批次都完成時需要更長的時間？

在上圖中，即使第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)

來源

2017-04-22 ssahu

當一個時代結束，因爲你是提供驗證數據，Keras必須對評估組評估模型，這就是發生在你的情況下，8分鐘。

來源

2017-04-22 20:19:51

感謝您的回答，它是否正常需要花費2分鐘才能完成超過4320個樣本的培訓，並驗證超過2.8k樣本需要6分鐘？培訓既能實現後向和前向支持，也需要前向支持。 – ssahu

@ssahu查看你的日誌一個紀元需要8分鐘，驗證數據的評估不是那8分鐘的一部分。 –

我正在通過'''nb_val_samples = nb_validation_samples'''而不是''nb_val_samples = nb_validation_samples // batch_size''''批量驗證''model.fit_generator'''，感謝您的洞察力，它幫我找到了問題。 – ssahu

Keras：爲什麼整個時期顯示所有批次都完成時需要更長的時間？

回答

相關問題