Commit eb0aa840 authored by Yuxin Wu's avatar Yuxin Wu

remove enqueue timeout (fix #324)

parent 01cab873
...@@ -188,20 +188,15 @@ class EnqueueThread(ShareSessionThread): ...@@ -188,20 +188,15 @@ class EnqueueThread(ShareSessionThread):
def run(self): def run(self):
with self.default_sess(): with self.default_sess():
sess = tf.get_default_session()
opt = tf.RunOptions()
opt.timeout_in_ms = 10000
try: try:
self.dataflow.reset_state() self.dataflow.reset_state()
while True: while True:
for dp in self.dataflow.get_data(): for dp in self.dataflow.get_data():
feed = dict(zip(self.placehdrs, dp)) feed = dict(zip(self.placehdrs, dp))
# print 'qsize:', self.sess.run([self.op, self.size_op], feed_dict=feed)[1] # print 'qsize:', self.sess.run([self.op, self.size_op], feed_dict=feed)[1]
sess.run(self.op, feed_dict=feed, options=opt) self.op.run(feed_dict=feed)
except (tf.errors.CancelledError, tf.errors.OutOfRangeError): except (tf.errors.CancelledError, tf.errors.OutOfRangeError):
pass pass
except tf.errors.DeadlineExceededError:
logger.exception("Deadline exceeded while waiting for data from the queue!")
except Exception: except Exception:
logger.exception("Exception in EnqueueThread:") logger.exception("Exception in EnqueueThread:")
finally: finally:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment