tensorflow - Colab 中的 ResourceExhausted 错误 - 使用动力学标签进行动作识别

我尝试在colab中使用此 Kinetics 标签进行动作识别。我参考这个链接

当我提供低于 2 MB 的输入视频时，该模型运行良好。但是，如果我给输入视频超过 2 MB，我会在几分钟后得到 ResourceExhausted 错误，我得到 GPU 内存使用量接近极限。即使我终止了笔记本并启动了新的笔记本，我也遇到了同样的错误。

标签： tensorflowactiongithub-pagesgoogle-colaboratory

正如错误所说，已达到硬件的物理限制。它需要比可用内存更多的 GPU 内存。

您可以通过减少模型的批量大小或调整输入视频序列的分辨率来防止这种情况。

或者，您可以尝试使用 Google 的云培训来获得额外的硬件资源，但它不是免费的。