make InputSource subclass methods private

1e7fa5f9 · Yuxin Wu · ef9e27a6 · 1e7fa5f9 · 1e7fa5f9
Commit 1e7fa5f9 authored Jul 11, 2017 by Yuxin Wu
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 95 additions and 54 deletions

tensorpack/callbacks/inference_runner.py tensorpack/callbacks/inference_runner.py +2 -1

tensorpack/train/input_source.py tensorpack/train/input_source.py +93 -53

No files found.
--- a/tensorpack/callbacks/inference_runner.py
+++ b/tensorpack/callbacks/inference_runner.py
@@ -270,7 +270,8 @@ class DataParallelInferenceRunner(InferenceRunnerBase):
                total -= nr_tower
            # take care of the rest
            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)
                pbar.update(1)
                total -= 1

--- a/tensorpack/train/input_source.py
+++ b/tensorpack/train/input_source.py