Skip to content

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

missing a keyword argument (fix #256)

parent 26d950fe
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/predict/base.py
View file @ f9fc329b
...@@ -137,7 +137,7 @@ class OfflinePredictor(OnlinePredictor): ...@@ -137,7 +137,7 @@ class OfflinePredictor(OnlinePredictor):
self.graph = config._maybe_create_graph() self.graph = config._maybe_create_graph()
with self.graph.as_default(): with self.graph.as_default():
input_placehdrs = config.model.get_reused_placehdrs() input_placehdrs = config.model.get_reused_placehdrs()
with TowerContext('', False): with TowerContext('', is_training=False):
config.model.build_graph(input_placehdrs) config.model.build_graph(input_placehdrs)
input_tensors = get_tensors_by_names(config.input_names) input_tensors = get_tensors_by_names(config.input_names)
......
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