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

refactor inference runners

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