【发布时间】:2020-01-30 03:03:34
【问题描述】:
如果我在 TFJS 中进行训练并且需要让我的笔记本电脑进入睡眠状态,那么当我唤醒笔记本电脑时训练会停止。控制台说:
WebGL: CONTEXT_LOST_WEBGL: loseContext: context lost
随后出现异常。
有没有办法避免这种情况?
【问题讨论】:
标签: tensorflow.js
如果我在 TFJS 中进行训练并且需要让我的笔记本电脑进入睡眠状态,那么当我唤醒笔记本电脑时训练会停止。控制台说:
WebGL: CONTEXT_LOST_WEBGL: loseContext: context lost
随后出现异常。
有没有办法避免这种情况?
【问题讨论】:
标签: tensorflow.js
由于 webgl 使用的内存增加,训练停止。代码中可能存在内存泄漏。
怎么办?
使用 tf.memory() 检查内存占用情况
清理所有未使用的张量
【讨论】: