Exemplos de Session em Python

Exemplo n.º 1

Arquivo: train_session.py Projeto: zhuhan1236/x-deeplearning

class SimpleSession(object):
    def __init__(self, hooks=None):
        env = current_env()
        if env is None:
            self._is_chief = True
        else:
            self._is_chief = env.is_chief
        if hooks is None:
            hooks = []
        self._hooks = hooks
        if self._is_chief:
            #self._hooks = self._hooks + [ChiefHook(), WorkerHook()]
            self._hooks = self._hooks + [ChiefHook()]
        else:
            self._hooks = self._hooks + [WorkerHook()]
        def take_priority(elem):
            return elem._priority
        self._hooks.sort(key=take_priority)
        self._session = Session(self._hooks)

    def run(self, v, run_option=None, run_statistic=None):
        return self._session.run(v, run_option, run_statistic)

    def end(self):
        return self._session.end()

Exemplo n.º 2

Arquivo: train_session.py Projeto: zscwind/x-deeplearning

 def __init__(self, hooks=None):
     env = current_env()
     if env is None:
         self._is_chief = True
     else:
         self._is_chief = env.is_chief
     if hooks is None:
         hooks = []
     self._hooks = hooks
     if self._is_chief:
         self._hooks = [ChiefHook(), WorkerHook()] + self._hooks
     else:
         self._hooks = [WorkerHook()] + self._hooks
     self._session = Session(self._hooks)

Exemplo n.º 3

Arquivo: train_session.py Projeto: zhuhan1236/x-deeplearning

 def __init__(self, hooks=None):
     env = current_env()
     if env is None:
         self._is_chief = True
     else:
         self._is_chief = env.is_chief
     if hooks is None:
         hooks = []
     self._hooks = hooks
     if self._is_chief:
         #self._hooks = self._hooks + [ChiefHook(), WorkerHook()]
         self._hooks = self._hooks + [ChiefHook()]
     else:
         self._hooks = self._hooks + [WorkerHook()]
     def take_priority(elem):
         return elem._priority
     self._hooks.sort(key=take_priority)
     self._session = Session(self._hooks)

Exemplo n.º 4

class SimpleSession(object):
    def __init__(self, hooks=None):
        env = current_env()
        if env is None:
            self._is_chief = True
        else:
            self._is_chief = env.is_chief
        if hooks is None:
            hooks = []
        self._hooks = hooks
        if self._is_chief:
            self._hooks = [ChiefHook(), WorkerHook()] + self._hooks
        else:
            self._hooks = [WorkerHook()] + self._hooks
        self._session = Session(self._hooks)

    def run(self, v, run_option=None, run_statistic=None, feed_dict=None):
        return self._session.run(v,
                                 run_option,
                                 run_statistic,
                                 feed_dict=feed_dict)

    def end(self):
        return self._session.end()