Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit 1e7fa5f9 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

make InputSource subclass methods private

parent ef9e27a6
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 95 additions and 54 deletions
tensorpack/callbacks/inference_runner.py
View file @ 1e7fa5f9
...@@ -270,7 +270,8 @@ class DataParallelInferenceRunner(InferenceRunnerBase): ...@@ -270,7 +270,8 @@ class DataParallelInferenceRunner(InferenceRunnerBase):
total -= nr_tower total -= nr_tower
# take care of the rest # take care of the rest
while total > 0: while total > 0:
feed = self._input_source.next_feed(cnt=1) # TODO XXX doesn't support remap
feed = self._input_source._next_feed(cnt=1)
self._hooked_sess.run(fetches=[], feed_dict=feed) self._hooked_sess.run(fetches=[], feed_dict=feed)
pbar.update(1) pbar.update(1)
total -= 1 total -= 1
......
tensorpack/train/input_source.py
View file @ 1e7fa5f9
This diff is collapsed.
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