首页 > 解决方案 > Tensorflow optimiser.compute_gradients 在第一步运行速度较慢

问题描述

我有一个用 tensorflow编写的复杂rnn模型,用于多 GPU训练。当我运行我的tf代码时,图形构建会在合理的时间内发生。

但是,当我调用sess.run运行图时,optimiser.compute_gradients 在第一步运行需要大量时间,但在后续步骤中会变得更快(大约快 10 倍)。我的数据集中的每个批次都具有相同的形状。

为什么会这样?

谢谢

标签: pythontensorflowdeep-learningrnnmulti-gpu

解决方案


推荐阅读