Method _register_hook

tensorpack/predict/feedfree.py:50–55

Args: hook (tf.train.SessionRunHook):

(self, hook)

            self._initialize_session()

    def _register_hook(self, hook):
        """
        Args:
            hook (tf.train.SessionRunHook):
        """
        self._hooks.append(hook)

    def _initialize_session(self):
        # init the session