{ | |
"init_args": [ | |
{ | |
"vocab_size": 12000, | |
"embedding_size": 384, | |
"hidden_size": 384, | |
"num_hidden_layers": 6, | |
"num_attention_heads": 6, | |
"intermediate_size": 1536, | |
"hidden_act": "gelu", | |
"hidden_dropout_prob": 0.1, | |
"attention_probs_dropout_prob": 0.1, | |
"max_position_embeddings": 512, | |
"type_vocab_size": 2, | |
"initializer_range": 0.02, | |
"pad_token_id": 0, | |
"rotary_value": false, | |
"init_class": "RoFormerModel" | |
} | |
], | |
"init_class": "RoFormerForPretraining" | |
} |