Commit 9bbdf94d authored by Yuxin Wu's avatar Yuxin Wu

Fix inference with feedinput

parent 6eb4b244
...@@ -116,7 +116,7 @@ class InferenceRunner(InferenceRunnerBase): ...@@ -116,7 +116,7 @@ class InferenceRunner(InferenceRunnerBase):
device (int): the device to use device (int): the device to use
""" """
if isinstance(input, DataFlow): if isinstance(input, DataFlow):
input = FeedInput(input, infinite=False) input = FeedInput(input, infinite=True) # TODO a better way to handle inference size
assert isinstance(input, InputSource), input assert isinstance(input, InputSource), input
self._tower_name = tower_name self._tower_name = tower_name
self._device = device self._device = device
......
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