tensorflow 队列管理之异步操作

阳光穿透心脏的1/2处 2022-05-13 15:34 126阅读 0赞

# -*- coding: utf-8 -*-
    """
    Created on Thu Oct 25 09:49:34 2018
    
    @author: Grey
    """
    import tensorflow as tf
    #模拟异步子线程存入样本，主线程读取样本
    #1.定义一个队列，100
    Q = tf.FIFOQueue(100,tf.float32)
    
    #2.定义子线程做的操作，循环+1 放入队列
    var = tf.Variable(0.0)#变量op
    #实现自增tf.assign_add()
    data=tf.assign_add(var,tf.constant(1.0))
    en_q=Q.enqueue(data)
    #3.定义队列管理器op，指定子线程的数量和操作
    qr=tf.train.QueueRunner(Q,enqueue_ops=[en_q]*2)
    
    
    
    #定义会话，执行op
    with tf.Session() as sess:
    #初始化var变量op
        tf.global_variables_initializer().run()
    #    正式开启子线程
        threads=qr.create_threads(sess,start=True)
    #    主线程，不断读取训练数据
        for i in range(300):
            print(sess.run(Q.dequeue()))
        pass

输出正确，但是还是会报错  
CancelledError (see above for traceback): Enqueue operation was cancelled  
         \[\[Node: fifo\_queue\_57\_enqueue = QueueEnqueue\[Tcomponents=\[DT\_FLOAT\], \_class=\["loc:@fifo\_queue\_57"\], timeout\_ms=-1, \_device="/job:localhost/replica:0/task:0/cpu:0"\](fifo\_queue\_57, AssignAdd\_31)\]\]

原因：主线程结束，意味着Session结束，资源释放，但是子线程还在运行，所以出错。

解决方案：线程协调器关闭子线程

![70][]

异步执行的关键是 创建子线程并及进行关闭

如下图为添加关闭子线程的异步执行的框架：

![70 1][]

代码：

# -*- coding: utf-8 -*-
    """
    Created on Thu Oct 25 09:49:34 2018
    
    @author: Grey
    """
    import tensorflow as tf
    
    Q = tf.FIFOQueue(1000,dtypes=tf.float32)
    
    var = tf.Variable(0.0)
    data=tf.assign_add(var,tf.constant(1.0))
    en_q=Q.enqueue(data)
    
    qr=tf.train.QueueRunner(Q,enqueue_ops=[en_q]*2)
    
    with tf.Session() as sess:
    
        tf.global_variables_initializer().run()
    
        coord=tf.train.Coordinator() 
        threads=qr.create_threads(sess,coord=coord,start=True)
    
    #    主线程，不断读取训练数据
        for i in range(100):
            print(sess.run(Q.dequeue()))
            
    #        回收子线程
        coord.request_stop()
        coord.join(threads)
        pass

输出结果正确且无任何报错：

![70 2][]

[70]: /images/20220504/40e1d9a405e4477da65863578128b621.png
[70 1]: /images/20220504/e224f53ba72c467b8a9a56ae849d790f.png
[70 2]: /images/20220504/1cdd5983beb540cc9919b25e9aaaade6.png