我的 Tensorflow神经网络在训练时逐渐变慢。每当我训练网络时,它的运行速度就会越来越慢。
经过30次左右的训练后,它变得缓慢得令人难以忍受,并且几乎无法使用。到第60次迭代时,程序停止响应。
我不认为这个神经网络是如此复杂。这是一个与Tensorflow结合在一起的简单的三层网络。
你们知道如何解决这个问题吗?
start
答案 0 :(得分:2)
这听起来像是内存问题。您不会在训练方法中删除l3或数组,也不会在预测方法中删除l3。我认为这不是问题的原因,因为无论如何,我相信无论如何,这应该被Python解释器丢弃。
您如何称呼这堂课?可能是因为您保留了内存中的输出,并且输出变得非常大。
答案 1 :(得分:0)
似乎每次调用train()
时,tf都在重建图形。尝试在train()
函数以及predict()
函数之外定义网络,损耗和训练师。