QueueRunner解决tensorflow框架下sess.run()读取数据卡住

Evan_Tech

于 2020-02-05 19:00:40 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow

本文链接：https://blog.csdn.net/yfren1123/article/details/104186823

本文介绍了如何利用Coordinator和QueueRunner在TensorFlow框架下解决sess.run()因数据读取卡住的问题。Coordinator负责管理并同步多线程，而QueueRunner则启动数据入队线程，确保训练数据被正确加载到内存序列中，避免程序等待。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

with tf.Session() as sess:
    summary_writer = tf.summary.FileWriter("../test_dir/")
    
    sess.run(tf.global_variables_initializer())
    sess.run(tf.l

最低0.47元/天解锁文章

200万优质内容无限畅学