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

Question

因为我有一个大数据集，我应该使用train_on_batch而不是model.fit，我想用reduceLROnplateau它来回调。我该怎么办？

Answer 1

您不需要仅仅因为您的数据集很大就使用 train_on_batch。您可以使用 model.fit。您需要做的是创建一个生成器，将数据分批提供给 model.fit。然后在 model.fit 中，您可以使用 reduceLROn 高原回调。如果您正在处理图像，您可以使用 ImaageDataGenerator 来生成批次。在一般情况下，您必须创建一个 python 生成器来生成批量数据。