def train_hooks(hook_context): del hook_context return [update_ops_hook.UpdateOpsHook()]
def train_hooks(): return [update_ops_hook.UpdateOpsHook()]