mindformers.core.build_context

mindformers.core.build_context(config: Union[dict, MindFormerConfig, TrainingArguments])[源代码]

基于config创建运行环境的context。

说明

当config类型是dict时，必须包含context和parallel属性。

参数：

config (Union[dict, MindFormerConfig, TrainingArguments]) - 初始化context的配置项，可以是字典类型，MindFormerConfig实例，TrainingArguments实例。

返回：

_Context，实例化后的context对象。

样例：

>>> from mindformers import build_context
>>> config = {'context': {'mode': 'GRAPH_MODE'}, 'parallel':{}}
>>> build_context(config=config)