recwizard.BaseConfig

The BaseConfig class adds a few features to the Huggingface PretrainedConfig class.

class recwizard.configuration_utils.BaseConfig(WEIGHT_DIMENSIONS=None, **kwargs)

The base config for all modules/pipelines.

__init__(WEIGHT_DIMENSIONS=None, **kwargs)

Parameters:

• WEIGHT_DIMENSIONS (dict, optional) – The dimension and dtype of module parameters. Used to initialize the parameters when they are not explicitly specified in module initialization. Defaults to None. See also recwizard.module_utils.BaseModule.prepare_weight().

• **kwargs – Additional parameters. Will be passed to the PretrainedConfig.__init__.