Function func_wrapper

guided_diffusion/logger.py:311–313

(*args, **kwargs)

    def decorator_with_name(func):
        def func_wrapper(*args, **kwargs):
            with profile_kv(n):
                return func(*args, **kwargs)

        return func_wrapper