Setup sessinit before creating session (fix #449)

98c3c7e1 · Yuxin Wu · 9268bc8c · 98c3c7e1
Commit 98c3c7e1 authored Oct 23, 2017 by Yuxin Wu
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tensorpack/predict/base.py tensorpack/predict/base.py +2 -1

No files found.
--- a/tensorpack/predict/base.py
+++ b/tensorpack/predict/base.py
@@ -163,7 +163,8 @@ class OfflinePredictor(OnlinePredictor):
            input_tensors = get_tensors_by_names(config.input_names)
            output_tensors = get_tensors_by_names(config.output_names)

+            config.session_init._setup_graph()
            sess = config.session_creator.create_session()
-            config.session_init.init(sess)
+            config.session_init._run_init(sess)
            super(OfflinePredictor, self).__init__(
                input_tensors, output_tensors, config.return_input, sess)