custom-mybertclassifier / configuration_my_bert_classifier.py
Capstone-lpx's picture
Upload MyBertClassifier
0c45ca3 verified
raw
history blame contribute delete
242 Bytes
from transformers import PretrainedConfig, BertModel
class MyBertClassifierConfig(PretrainedConfig):
model_type = "my_bert_classifier"
def __init__(
self,
**kwargs,
):
super().__init__(**kwargs)