标签: tensorflow elmo
我正在训练Elmo嵌入到我的数据中(300000个句子,最大长度为150个字符)。我的系统具有GPU 1060和16 GB内存。问题是在training the model期间,我将代码设置为使用GPU,并且可以看到它使用GPU,但是问题是,当我在训练期间看到可用内存时,可以看到内存减少到100 mb!我更改了tensorflow设置,解决了ram的问题,但过一会儿我发现代码又停止了,没有错误消息。如何找到GPU ram或系统RAM的问题?
Elmo