Function func_with_graph_arg

tensorpack/utils/argtools.py:51–53

(*args, **kwargs)

    @memoized
    def func_with_graph_arg(*args, **kwargs):
        kwargs.pop(GRAPH_ARG_NAME)
        return func(*args, **kwargs)

    @functools.wraps(func)
    def wrapper(*args, **kwargs):