我正在研究图像分类模型。我的模型是使用 CNN、tensorflow 和 Keras 库构建的。
我正在尝试在谷歌 Colab 中执行我的 CNN 多分类模型，并且我正在从我的谷歌驱动器（装载驱动器命令）加载数据（图像数据）。一切都非常顺利，直到 model.fit，此时模型执行在 Google Colab 中是无限期的（永无止境的）。然而，相同的模型在 Jupiter Notebook 中运行良好。
那么，谁能告诉我发生了什么？
Google Colab 中的 model.fit 无限期执行
数据挖掘
喀拉斯
张量流
美国有线电视新闻网
2022-02-22 16:30:01
1个回答
我已经解决了这个问题。
如果其他人处理类似情况，解决方案是为训练和验证图像分配一个 batch_size。然后在 model.fit 中添加steps_per_epochand作为参数。validation_set
steps_per_epochs= number_of_training_images/batch_size
validation_set = number_of_test_images/batch_size