mindformers.tools.register.MindFormerModuleType

查看源文件
class mindformers.tools.register.MindFormerModuleType[源代码]

MindFormers模块类型的枚举类。其中包括：

枚举名

值

含义

CALLBACK

'callback'

回调函数

CONFIG

'config'

配置

CONTEXT

'context'

上下文

DATASET

'dataset'

数据集

DATASET_LOADER

'dataset_loader'

数据集加载器

DATASET_SAMPLER

'dataset_sampler'

数据集采样器

DATA_HANDLER

'data_handler'

数据处理器

ENCODER

'encoder'

编码器

FEATURE_EXTRACTOR

'feature_extractor'

特征提取器

LOSS

'loss'

损失函数

LR

'lr'

学习率

MASK_POLICY

'mask_policy'

掩码策略

METRIC

'metric'

评测指标

MODELS

'models'

模型

OPTIMIZER

'optimizer'

优化器

PIPELINE

'pipeline'

pipeline接口

PROCESSOR

'processor'

处理器

TOKENIZER

'tokenizer'

分词器

TOOLS

'tools'

工具

TRAINER

'trainer'

Trainer接口

TRANSFORMS

'transforms'

数据转换

WRAPPER

'wrapper'

包装函数

样例：

>>> from mindformers.tools import MindFormerModuleType
>>>
>>> print(MindFormerModuleType.MODULES)
modules