mindformers.tools.MindFormerConfig

class mindformers.tools.MindFormerConfig(*args, **kwargs)

一个配置的类，继承于Python的dict类。可以解析来自yaml文件或dict实例的配置参数。

参数：

• args (Any) - 可扩展参数列表，可以是yaml配置文件路径或配置字典。

• kwargs (Any) - 可扩展参数字典，可以是yaml配置文件路径或配置字典。

返回：

一个类的实例。

样例：

>>> from mindformers.tools import MindFormerConfig
>>>
>>> # test.yaml:
>>> #     a:1
>>> cfg = MindFormerConfig('./test.yaml')
>>> cfg.a
1
>>> cfg = MindFormerConfig(**dict(a=1, b=dict(c=[0,1])))
>>> cfg.b
{'c': [0, 1]}

merge_from_dict(options)

将配置选项合并入配置中。

参数：

• options (dict) - 需要合并的配置选项。

样例：

>>> from mindformers.tools import MindFormerConfig
>>>
>>> options = {'model.arch': 'LlamaForCausalLM'}
>>> cfg = MindFormerConfig(**dict(model=dict(model_config=dict(type='LlamaConfig'))))
>>> cfg.merge_from_dict(options)
>>> print(cfg)
{'model': {'model_config': {'type': 'LlamaConfig'}, 'arch': 'LlamaForCausalLM'}}