Skip to content

S
seminar-breakout
Commit 7290319f authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

bugfix in DataParallelInferenceRunner

parent 59160bcf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tensorpack/callbacks/inference_runner.py
View file @ 7290319f
...@@ -260,4 +260,5 @@ class DataParallelInferenceRunner(InferenceRunnerBase): ...@@ -260,4 +260,5 @@ class DataParallelInferenceRunner(InferenceRunnerBase):
feed = self._input_source.next_feed(cnt=1) 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
summary_inferencer(self.trainer, self.infs) summary_inferencer(self.trainer, self.infs)
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