fit() 得到了一个意外的关键字参数“生成器”答案

【问题标题】：fit() got an unexpected keyword argument 'generator'fit() 得到了一个意外的关键字参数“生成器”
【发布时间】：2021-03-21 19:06:45
【问题描述】：

epochs = 150
callbacks_list=[ModelCheckpoint(save_best_only=False,filepath=checkpoint_path),TensorBoard(log_dir='logs')]
history = model.fit(generator=gen_tr,steps_per_epoch=200,
                                  epochs=epochs,
                                  validation_data=gen_val,
                                  validation_steps=1,
                                  callbacks = callbacks_list )

Error message- TypeError : fit() got an unexpected keyword argument 'generator'

在文档中写到 Model.fit 现在支持生成器，但仍然显示错误。

【问题讨论】：

应该是model.fit(gen_tr...)，没有关键字generator。
谢谢。解决了。

标签： tensorflow tf.keras

【解决方案1】：

来自 cmets

epochs = 150
callbacks_list=[ModelCheckpoint(save_best_only=False,filepath=checkpoint_path),TensorBoard(log_dir='logs')]
history = model.fit(gen_tr,steps_per_epoch=200,
                                  epochs=epochs,
                                  validation_data=gen_val,
                                  validation_steps=1,
                                  callbacks = callbacks_list )

（转述自 Frightera）

更多详情可以参考model.fit。

【讨论】：

【解决方案2】：

Model.fit() 接收生成器。但删除关键字生成器。

model.fit(generate_batch(X_train, y_train, batch_size = batch_size),...)

【讨论】：