我通过使用fit_genertor解决了这个问题。我在这里找到了解决方案。我应用了@ Dat-
Nguyen的解决方案。
您只需创建两个迭代器，一个用于训练，一个用于验证，然后创建您自己的生成器，即可从数据集中提取批次，并以（batch_data，batch_labels）的形式提供数据。最后，在model.fit_generator中，您将传递train_generator和validation_generator。
我通过使用fit_genertor解决了这个问题。我在这里找到了解决方案。我应用了@ Dat-
Nguyen的解决方案。
您只需创建两个迭代器，一个用于训练，一个用于验证，然后创建您自己的生成器，即可从数据集中提取批次，并以（batch_data，batch_labels）的形式提供数据。最后，在model.fit_generator中，您将传递train_generator和validation_generator。
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
