Tensorflow while_loop用于训练#2

时间:2016-10-20 17:21:12

标签: multithreading tensorflow

Here我问过如何通过 while_loop 进行培训来解决开销问题(通过调用只有一次运行,可以多次评估 train_op )。之后,我创建了4个线程并为每个线程运行一个 while_loop 以进行并行优化。 TensorFlow中是否存在用于此类并行优化的本机机制?

我使用Ftrl优化器。

谢谢!

编辑:

在我的情况下,我有大数据集,我在主线程中逐渐读取并入队到FIFOQueue。我使用批量优化和小批量的一个优化步骤(理想的只有一个元素)花费很少的时间(我使用线性模型),因为我想在一个运行调用中执行所有优化步骤,而不返回到每个步骤的python解释器(因为开销问题)。现在我调用run次数和线程数一样多。

0 个答案:

没有答案