Skip to content

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

fix a similar issue like #1020

parent b81758c6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/predict/multigpu.py
View file @ 79148350
...@@ -114,7 +114,8 @@ class DataParallelOfflinePredictor(OnlinePredictor): ...@@ -114,7 +114,8 @@ class DataParallelOfflinePredictor(OnlinePredictor):
input_tensors.extend(h.get_tensors(config.input_names)) input_tensors.extend(h.get_tensors(config.input_names))
output_tensors.extend(h.get_tensors(config.output_names)) output_tensors.extend(h.get_tensors(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(DataParallelOfflinePredictor, self).__init__( super(DataParallelOfflinePredictor, 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