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

  • Z时代
  • 2024-01-10
  • 分类：问答

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？

回答：

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（在fit_generator弃用之后） 的全部内容， 来源链接： utcz.com/qa/426770.html

回到顶部