如何使用支持生成器的Model.fit在fit_generator弃用之后

编程问答 2021-10-20

问题描述

Model.fit_generator从目前在 tensorflow 2.1.0开始不推荐使用。您可以在这里找到tf-2.1.0-rc1的文档：https ://www.tensorflow.org/versions/r2.1/api_docs/python/tf/keras/Model#fit

如您所见，的第一个参数Model.fit可以使用生成器，因此只需将其传递给生成器即可。

解决方法

Model.fit_generator在tensorflow中使用时，我收到了此弃用警告：

WARNING:tensorflow: Model.fit_generator (from tensorflow.python.keras.engine.training) is deprecated and will be removed in a future version.
Instructions for updating:
Please use Model.fit,which supports generators.

我怎么用Model.fit代替Model.fit_generator？

python