【发布时间】:2020-08-24 23:27:16
【问题描述】:
我现在正在 Google Colab 上进行模型训练。就在一分钟前,本地系统消息弹出,说内存即将耗尽(同时我没有运行任何东西)。
现在我的内存使用量是 40%,但是每次打开 Colab 时,本地内存使用量峰值达到 80 并且不断增加。 Colab 页面冻结,因此我无法停止当前运行时。
这怎么可能? Colab 不应该是一个在线环境吗?可以和tqdm 包有关吗?因为以前的版本运行良好,所以这种情况只有在我添加tqdm 以在屏幕上显示进度条时才会出现。
【问题讨论】:
-
类似的症状,使用 PyTorch 和 PyTorch Lightning。对我来说,似乎每个验证步骤都会在笔记本中产生一个新的 tqdm 条,以及其他问题。我用压缩器绕过了 Rey 之类的问题,通过将 progress_bar_refresh_rate 减少(实际上增加)到 50 并使用 check_val_every_n_epoch 将完全验证设置为可能每 50 个 epoch 一次。仍然不知道根本原因是什么。
标签: python-3.x google-colaboratory tqdm