rubert-siamese / configuration_siamese.py
AlanRobotics's picture
Upload model
cf14450
raw
history blame
267 Bytes
from transformers import PretrainedConfig
class SiameseConfig(PretrainedConfig):
model_type = "siamese"
def __init__(
self,
**kwargs):
super().__init__()
siamese_config = SiameseConfig()
siamese_config.save_pretrained('siamse_nn')