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')