问基于Tensorflow的迁移学习(MobileNet)

EN

如果您按下面的方式更新最后一行，则5个周期大约需要40秒。

从…

model.fit_generator(train_generator, epochs=5, validation_data=valid_generator)

至

model.fit_generator(train_generator, epochs=5, steps_per_epoch=len(train_generator), validation_data=valid_generator, validation_steps=len(valid_generator))

当model.fit的输入是生成器时，请检查对期望的描述。因此，当“steps_per_epoch”为空时，该时代将运行到输入数据集耗尽为止。因此，如果您的数据集是无限重复数据集，则生成器将无限地运行。

steps_per_epoch:整数还是零。在声明一个时代完成并开始下一个阶段之前，步骤总数(批样本)。当使用输入张量(如TensorFlow数据张量)进行训练时，默认值不等于数据集中的样本数除以批处理大小，如果无法确定，则为1。如果x是tf.data数据集，而“steps_per_epoch”为None，则该时代将运行到输入数据集耗尽为止。在传递无限重复数据集时，必须指定steps_per_epoch参数。数组输入不支持此参数。