Skip to content

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

Setup sessinit before creating session (fix #449)

parent 9268bc8c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/predict/base.py
View file @ 98c3c7e1
...@@ -163,7 +163,8 @@ class OfflinePredictor(OnlinePredictor): ...@@ -163,7 +163,8 @@ class OfflinePredictor(OnlinePredictor):
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)
config.session_init._setup_graph()
sess = config.session_creator.create_session() sess = config.session_creator.create_session()
config.session_init.init(sess) config.session_init._run_init(sess)
super(OfflinePredictor, self).__init__( super(OfflinePredictor, self).__init__(
input_tensors, output_tensors, config.return_input, sess) input_tensors, output_tensors, config.return_input, sess)
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