为什么我开始训练后,我的Tensorflow神经网络的动作这么慢?

时间:2018-11-24 02:23:42

标签: python tensorflow machine-learning neural-network

我的 Tensorflow神经网络在训练时逐渐变慢。每当我训练网络时,它的运行速度就会越来越慢。

经过30次左右的训练后,它变得缓慢得令人难以忍受,并且几乎无法使用。到第60次迭代时,程序停止响应。

我不认为这个神经网络是如此复杂。这是一个与Tensorflow结合在一起的简单的三层网络。

你们知道如何解决这个问题吗?

start

2 个答案:

答案 0 :(得分:2)

这听起来像是内存问题。您不会在训练方法中删除l3或数组,也不会在预测方法中删除l3。我认为这不是问题的原因,因为无论如何,我相信无论如何,这应该被Python解释器丢弃。

您如何称呼这堂课?可能是因为您保留了内存中的输出,并且输出变得非常大。

答案 1 :(得分:0)

似乎每次调用train()时,tf都在重建图形。尝试在train()函数以及predict()函数之外定义网络,损耗和训练师。

相关问题