使用GPU在Colab(Tensorflow)上拟合模型花费的时间过长

jogvjijk  于 2022-11-16  发布在  其他
关注(0)|答案(1)|浏览(149)

我正在学习TensorFlow的课程。在尝试使用GPU在Colab上拟合模型时,它显示了ETA:96:45:07。通常不会花这么长时间。数据集比我习惯处理的要大,但根据video

,不应该花这么长时间
我在试着拟合模型,但我没想到它会那么长。

y1aodyip

y1aodyip1#

首先,欢迎来到Stack Overflow:D
您看到这种情况的原因之一是,在第3行中,您使用了整个train_data作为steps_per_epoch。这意味着,对于3个epoch中的每一个,您都使用了整个train数据长度来通过数据集。
你能展示更多你的笔记本或链接到你用来编码的视频吗?这样我就能得到更多的上下文。

**旁注:**一般来说,如果您可以访问TPU,它的使用速度会比GPU快,但最终它归结为您如何尝试适合您的模型。

相关问题