Commit 7db397d7 authored by Yuxin Wu's avatar Yuxin Wu

refactor inference runners

parent 3bc0bed2
This diff is collapsed.
......@@ -80,6 +80,7 @@ class ModelDesc(object):
for v in input_vars:
tf.add_to_collection(INPUTS_KEY, v.dumps())
ret = []
with tf.name_scope(None): # clear any name scope it might get called in
for v in input_vars:
placehdr_f = tf.placeholder if not v.sparse else tf.sparse_placeholder
ret.append(placehdr_f(
......
......@@ -85,6 +85,7 @@ class FeedfreeInput(InputData):
pass
# TODO enqueu_many? https://github.com/tensorflow/tensorflow/issues/7817#issuecomment-282053155
class EnqueueThread(ShareSessionThread):
def __init__(self, queue, ds, input_placehdrs):
super(EnqueueThread, self).__init__()
......
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