Class CallbackToHook

tensorpack/callbacks/hooks.py:17–36

Hooks are less powerful than callbacks so the conversion is incomplete. It only converts the ``before_run/after_run`` calls. This is only for internal implementation of ``before_run/after_run`` callbacks. You shouldn't need to use this.

class CallbackToHook(tfv1.train.SessionRunHook):
    """
    Hooks are less powerful than callbacks so the conversion is incomplete.
    It only converts the ``before_run/after_run`` calls.

    This is only for internal implementation of
    ``before_run/after_run`` callbacks.
    You shouldn't need to use this.
    """

    def __init__(self, cb):
        self._cb = cb

    @HIDE_DOC
    def before_run(self, ctx):
        return self._cb.before_run(ctx)

    @HIDE_DOC
    def after_run(self, ctx, vals):
        self._cb.after_run(ctx, vals)


class HookToCallback(Callback):