Create the session and set `self.sess`. Call `self.initiailize_hooks()` Finalize the graph. It must be called after callbacks are setup. Args: session_creator (tf.train.SessionCreator): session_init (sessinit.SessionInit):
(self, session_creator, session_init)
| 210 | |
| 211 | @call_only_once |
| 212 | def initialize(self, session_creator, session_init): |
| 213 | """ |
| 214 | Create the session and set `self.sess`. |
| 215 | Call `self.initiailize_hooks()` |
| 216 | Finalize the graph. |
| 217 | |
| 218 | It must be called after callbacks are setup. |
| 219 | |
| 220 | Args: |
| 221 | session_creator (tf.train.SessionCreator): |
| 222 | session_init (sessinit.SessionInit): |
| 223 | """ |
| 224 | assert isinstance(session_creator, tfv1.train.SessionCreator), session_creator |
| 225 | assert isinstance(session_init, SessionInit), session_init |
| 226 | session_init._setup_graph() |
| 227 | |
| 228 | logger.info("Creating the session ...") |
| 229 | |
| 230 | self.sess = session_creator.create_session() |
| 231 | self.initialize_hooks() |
| 232 | |
| 233 | if self.is_chief: |
| 234 | logger.info("Initializing the session ...") |
| 235 | session_init._run_init(self.sess) |
| 236 | else: |
| 237 | if not isinstance(session_init, JustCurrentSession): |
| 238 | logger.warn("This is not a chief worker, 'session_init' was ignored!") |
| 239 | |
| 240 | self.sess.graph.finalize() |
| 241 | logger.info("Graph Finalized.") |
| 242 | |
| 243 | @call_only_once |
| 244 | def initialize_hooks(self): |
no test coverage detected