Skip to content

S
seminar-breakout
Commit 5b5d203f authored by AprliRainkun's avatar AprliRainkun Committed by Yuxin Wu
Browse files
Options

fix predictor build graph (#459)

parent 2b41edf7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/predict/base.py
View file @ 5b5d203f
...@@ -158,7 +158,7 @@ class OfflinePredictor(OnlinePredictor): ...@@ -158,7 +158,7 @@ class OfflinePredictor(OnlinePredictor):
input = PlaceholderInput() input = PlaceholderInput()
input.setup(config.model.get_inputs_desc()) input.setup(config.model.get_inputs_desc())
with TowerContext('', is_training=False): with TowerContext('', is_training=False):
config.model.build_graph(input) config.model.build_graph(input.get_input_tensors())
input_tensors = get_tensors_by_names(config.input_names) input_tensors = get_tensors_by_names(config.input_names)
output_tensors = get_tensors_by_names(config.output_names) output_tensors = get_tensors_by_names(config.output_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