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

Question

我尝试使用 colab 中的Kinetics标签进行动作识别。 I refer this link

当我提供2 MB以下的输入视频时，此模型运行良好。但是，如果我给输入的视频超过2 MB，则几分钟后我收到ResourceExhausted错误，我的GPU内存使用率已接近极限。 甚至我终止笔记本并启动新笔记本，我也遇到同样的错误。

Answer 1

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

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

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

https://cloud.google.com/tpu/