bugfix in DataParallelInferenceRunner

7290319f · Yuxin Wu · 59160bcf · 7290319f
Commit 7290319f authored May 26, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

No files found.
--- a/tensorpack/callbacks/inference_runner.py
+++ b/tensorpack/callbacks/inference_runner.py
@@ -260,4 +260,5 @@ class DataParallelInferenceRunner(InferenceRunnerBase):
                feed = self._input_source.next_feed(cnt=1)
                self._hooked_sess.run(fetches=[], feed_dict=feed)
                pbar.update(1)
+                total -= 1
        summary_inferencer(self.trainer, self.infs)