Tensorflow seq2seq,桶使用太多内存

时间:2016-04-27 13:07:46

标签: tensorflow

我有这个代码:http://pastebin.com/7kcFekKG主要是从tensorflow的单元测试中复制而来,用桶创建一个seq2seq注意模型。如果我让测试代码运行,它工作正常,但如果我将序列长度增加到pastebin版本中显示的值,它使用超过8GB的内存,这对于具有96个神经元的rnn不是很好。特别是因为我想增加神经元的数量。有什么东西我做错了导致tf使用这么多内存吗?我已经尝试使用不同的aggegation方法但调试显示在第47行创建网络本身时内存消耗

0 个答案:

没有答案
相关问题