{ | |
"init_args": [ | |
{ | |
"attention_probs_dropout_prob": 0.1, | |
"hidden_act": "relu", | |
"hidden_dropout_prob": 0.1, | |
"hidden_size": 8, | |
"intermediate_size": 8, | |
"initializer_range": 0.02, | |
"max_position_embeddings": 512, | |
"num_attention_heads": 2, | |
"num_hidden_layers": 2, | |
"type_vocab_size": 2, | |
"vocab_size": 30522, | |
"pad_token_id": 0, | |
"init_class": "BertModel" | |
} | |
], | |
"init_class": "BertForMaskedLM" | |
} |