在使用Colab和fastai进行训练时遇到问题(首个epoch耗时超过4小时)

我在使用Colab和fastai进行训练时遇到了问题(首个epoch耗时超过4小时)。仅首个epoch就花费了大约5小时,这在使用Google Colab时是没有效率的,因为GPU的使用有限制。我使用的是’efficientnet-b4’模型,经过数据增强后有大约52,000张训练照片和11,000张验证照片。

问题如图所示:

Epoch训练时间


回答:

Colab从Google Drive获取数据需要很长时间,因为数据没有被高效索引,这使得首个epoch非常漫长。对于大型数据集,我建议使用Google云存储并从那里导入数据。

Related Posts

Keras Dense层输入未被展平

这是我的测试代码: from keras import…

无法将分类变量输入随机森林

我有10个分类变量和3个数值变量。我在分割后直接将它们…

如何在Keras中对每个输出应用Sigmoid函数?

这是我代码的一部分。 model = Sequenti…

如何选择类概率的最佳阈值?

我的神经网络输出是一个用于多标签分类的预测类概率表: …

在Keras中使用深度学习得到不同的结果

我按照一个教程使用Keras中的深度神经网络进行文本分…

‘MatMul’操作的输入’b’类型为float32,与参数’a’的类型float64不匹配

我写了一个简单的TensorFlow代码,但不断遇到T…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注