tensorflow - flowers_tf_lite.ipynb - 训练模型

Question

我在flowers_tf_lite.ipynb测试示例笔记本，当我进入训练单元时，它没有运行。

好像永远在无穷大

训练模型 - 未知

标签： tensorflowgoogle-colaboratory

Answer 1

@Pedro Asseiro 实际上，迭代器会动态生成数据。因此，数据集迭代器的长度在您至少迭代一次之前是未知的。您可以将 steps_per_epoch 参数传递给model.fit_generator如下所示。然后，它会按您的预期打印。

with tf.device("/device:GPU:0"):
  history = model.fit_generator(train_generator, steps_per_epoch=len(train_generator),
                    epochs=epochs, 
                    validation_data=val_generator,validation_steps=len(val_generator))

编辑1：

@Pedro Asseiro @gokul_uf 我根据您的评论更新了上述代码。您需要添加steps_per_epoch以及validation_steps. 我做的另一个改变是model.fit在 GPU 下运行。model.fit此外，我在微调部分做了完全相同的事情。

with tf.device("/device:GPU:0"):
  history_fine = model.fit_generator(train_generator, steps_per_epoch=len(train_generator),
                         epochs=5,
                         validation_data=val_generator,validation_steps=len(val_generator))

请在此处查看 GitHub 要点。谢谢！

请让我知道是否有任何问题。谢谢！