File size: 275 Bytes
ebfd652 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from transformers import PretrainedConfig
class SiameseConfig(PretrainedConfig):
model_type = "siamese"
def __init__(
self,
**kwargs):
super().__init__(**kwargs)
siamese_config = SiameseConfig()
siamese_config.save_pretrained('siamse_nn') |