We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 318ed29 commit 91d0a1fCopy full SHA for 91d0a1f
paddlenlp/transformers/ernie_m/configuration.py
@@ -14,6 +14,8 @@
14
""" ERNIE-M model configuration"""
15
from __future__ import annotations
16
17
+from typing import Dict
18
+
19
from ..configuration_utils import PretrainedConfig
20
21
__all__ = ["ERNIE_M_PRETRAINED_INIT_CONFIGURATION", "ErnieMConfig", "ERNIE_M_PRETRAINED_RESOURCE_FILES_MAP"]
@@ -142,6 +144,7 @@ class ErnieMConfig(PretrainedConfig):
142
144
>>> configuration = model.config
143
145
```"""
146
model_type = "ernie_m"
147
+ attribute_map: Dict[str, str] = {"num_classes": "num_labels", "dropout": "classifier_dropout"}
148
pretrained_init_configuration = ERNIE_M_PRETRAINED_INIT_CONFIGURATION
149
150
def __init__(
0 commit comments