utils¶
-
fn_args_to_dict
(func, *args, **kwargs)[source]¶ Inspect function
func
and its arguments for running, and extract a dict mapping between argument names and keys.
-
class
InitTrackerMeta
(name, bases, attrs)[source]¶ Bases:
pybind11_builtins.pybind11_type
This metaclass wraps the
__init__
method of a class to addinit_config
attribute for instances of that class, andinit_config
use a dict to track the initial configuration. If the class has_wrap_init
method, it would be hooked after__init__
and called as_wrap_init(self, init_fn, init_args)
. Since InitTrackerMeta would be used as metaclass for pretrained model classes, which always are Layer andtype(Layer)
is nottype
, thus usetype(Layer)
rather thantype
as base class for it to avoid inheritance metaclass conflicts.-
static
init_and_track_conf
(init_func, help_func=None)[source]¶ wraps
init_func
which is__init__
method of a class to addinit_config
attribute for instances of that class. :param init_func: It should be the__init__
method of a class. :type init_func: callable :param help_func: If provided, it would be hooked afterinit_func
and called as_wrap_init(self, init_func, *init_args, **init_args)
. Default None.- Returns
the wrapped function
- Return type
function
-
static