from transformers import PretrainedConfig, BertModel | |
class MyBertClassifierConfig(PretrainedConfig): | |
model_type = "my_bert_classifier" | |
def __init__( | |
self, | |
**kwargs, | |
): | |
super().__init__(**kwargs) |
from transformers import PretrainedConfig, BertModel | |
class MyBertClassifierConfig(PretrainedConfig): | |
model_type = "my_bert_classifier" | |
def __init__( | |
self, | |
**kwargs, | |
): | |
super().__init__(**kwargs) |