| 35 | # pprint([(k.name, k.device) for k in tf.local_variables()]) |
| 36 | |
| 37 | class _Creator(tf.train.SessionCreator): |
| 38 | def create_session(self): |
| 39 | if is_chief: |
| 40 | return sm.prepare_session(master=server.target, init_op=init_op) |
| 41 | else: |
| 42 | tf.logging.set_verbosity(tf.logging.INFO) # print message about uninitialized vars |
| 43 | ret = sm.wait_for_session(master=server.target) |
| 44 | tf.logging.set_verbosity(tf.logging.WARN) |
| 45 | return ret |
| 46 | |
| 47 | return _Creator() |
no outgoing calls
no test coverage detected
searching dependent graphs…