fix a bug cause Graph is finalized and cannot be modified like #449 (#1020)

b81758c6 · Hao Feng · Yuxin Wu · dda6fd53 · b81758c6
Commit b81758c6 authored Dec 19, 2018 by Hao Feng Committed by Yuxin Wu Dec 19, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

No files found.
--- a/tensorpack/predict/multigpu.py
+++ b/tensorpack/predict/multigpu.py
@@ -45,8 +45,9 @@ class MultiTowerOfflinePredictor(OnlinePredictor):
                    config.tower_func(*input.get_input_tensors())
                    handles.append(config.tower_func.towers[-1])

+            config.session_init._setup_graph()
            self.sess = config.session_creator.create_session()
-            config.session_init.init(self.sess)
+            config.session_init._run_init(self.sess)

            for h in handles:
                input_tensors = h.get_tensors(config.input_names)