首頁猿問如何在 keras...

如何在 keras 中使用帶有“train_on_batch”

Python

料青山看我應(yīng)如是 2023-05-16 16:44:21

因為我有一個大數(shù)據(jù)集，我應(yīng)該使用train_on_batch而不是model.fit，我想用reduceLROnplateau它來回調(diào)。我該怎么辦？

查看完整描述

1 回答

PIPIONE

TA貢獻(xiàn)1829條經(jīng)驗獲得超9個贊

您不需要僅僅因為您的數(shù)據(jù)集很大就使用 train_on_batch。您可以使用 model.fit。您需要做的是創(chuàng)建一個生成器，將數(shù)據(jù)分批提供給 model.fit。然后在 model.fit 中，您可以使用 reduceLROn 高原回調(diào)。如果您正在處理圖像，您可以使用 ImaageDataGenerator 來生成批次。在一般情況下，您必須創(chuàng)建一個 python 生成器來生成批量數(shù)據(jù)。

反對回復(fù) 2023-05-16