2018年9月26日
tensorflow训练过程中内存溢出
摘要： 罪魁祸首是训练过程中给模型传值时的如下语句： 而其中函数seq2embeded（）中用到了tensorflow的运算： 这两句会增加graph节点，使得图在训练过程中不断增大，就会不断消耗内存。 教训： 训练过程中，能用numpy处理的运算尽量不用tensorflow中的运算。 如果必须用tenso 阅读全文
posted @ 2018-09-26 11:52 wxiaoli 阅读(2676) 评论(0) 推荐(0) 编辑