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

Fix inference with feedinput

parent 6eb4b244
......@@ -116,7 +116,7 @@ class InferenceRunner(InferenceRunnerBase):
device (int): the device to use
"""
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
self._tower_name = tower_name
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