File size: 340 Bytes
3a9e97b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from transformers.models.bert.configuration_bert import BertConfig
class BertHashConfig(BertConfig):
"""
Extension of Bert configuration to add projections parameter.
"""
model_type = "bert_hash"
def __init__(self, projections=5, **kwargs):
super().__init__(**kwargs)
self.projections = projections
|