Commit a28f30e3 authored by Yuxin Wu's avatar Yuxin Wu

give deque a name in queueinputtrainer

parent 175bc41a
......@@ -124,7 +124,8 @@ class QueueInputTrainer(Trainer):
def _get_model_inputs(self):
""" Dequeue a datapoint from input_queue and return"""
ret = self.input_queue.dequeue()
ret = self.input_queue.dequeue(name='input_deque')
print ret
if isinstance(ret, tf.Tensor): # only one input
ret = [ret]
assert len(ret) == len(self.input_vars)
......
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