Skip to content

S
seminar-breakout
Commit 545fc709 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix DataParallelInferenceRunner (#365)

parent 2e7dd9c7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/graph_builder/input_source.py
View file @ 545fc709
...@@ -134,7 +134,7 @@ class DataParallelFeedInput(FeedInput): ...@@ -134,7 +134,7 @@ class DataParallelFeedInput(FeedInput):
# build a list of placeholders for each tower # build a list of placeholders for each tower
self._placehdrs_per_tower.append( self._placehdrs_per_tower.append(
[v.build_placeholder(prefix=tname + '/') for v in inputs]) [v.build_placeholder(prefix=tname + '/') for v in inputs])
self._cb = self._DataParallelFeedCallback(self._repeat_ds, self._placehdrs_per_tower) self._cb = self._DataParallelFeedCallback(self._iter_ds, self._placehdrs_per_tower)
self.reset_state() self.reset_state()
def _get_input_tensors(self): def _get_input_tensors(self):
......
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