def run(self, v, run_option=None, run_statistic=None): current_graph().unfinalize() try: if self._session is None: self._restart_client() self._session = SimpleSession(self._hooks) return self._session.run(v, run_option, run_statistic) except (PsError) as e: print( 'An error was raised. This may be due to a preemption in ' 'a connected worker or parameter server. The current ' 'session will be closed and a new session will be ' 'created. Error: %s', str(e)) time.sleep(5) self._session = None except OutOfRange: self._finish = True
def __init__(self, outs, g=None, weak_variable=False, slow_start=1000): if g is None: g = current_graph() self._g = g self._outs = _flatten_outs(outs) self._real_outs = [] self._weak_variable = weak_variable self._slow_start = slow_start self._step = False self._build() self._run_time = 0
def __init__(self): self._ckpt_dir = xdl.get_ckpt_dir() if self._ckpt_dir is None: raise ValueError('must specify ckpt_dir arg in cmdline') self._graph_def = _graphdef_to_pb(current_graph()._graph_def)
def enqueue_start(self): enqueue = xdl.enqueue_op(tensors=self._enqueue_tensors, names=";".join(self._names)) current_graph().execute_loop(enqueue) xdl.Graph._current_graph.pop()
def __init__(self, ckpt_dir=None, tf_graph_name=None): self._ckpt_dir = ckpt_dir if self._ckpt_dir is None: self._ckpt_dir = get_ckpt_dir() self._graph_def = _graphdef_to_pb(current_graph()._graph_def) self._tf_graph_name = tf_graph_name
def __init__(self, ckpt_dir=None): self._ckpt_dir = ckpt_dir self._graph_def = _graphdef_to_pb(current_graph()._graph_def)
def feed_output_list(self): ops = current_graph().nodes() for op in ops.values(): for i in range(len(op.inputs)): self._output_list[op.inputs[i]].append((op, i))