RoBERTa-mini / config.json
dilip025's picture
Update config.json
6a18d5d verified
raw
history blame
506 Bytes
{
"architectures": ["RobertaForSentimentClassification"],
"model_type": "roberta-sentiment",
"vocab_size": 30000,
"hidden_size": 512,
"num_attention_heads": 8,
"num_hidden_layers": 6,
"intermediate_size": 2048,
"max_position_embeddings": 128,
"num_labels": 5,
"hidden_dropout_prob": 0.1,
"auto_map": {
"AutoConfig": "modeling_roberta_sentiment.RobertaSentimentConfig",
"AutoModelForSequenceClassification": "modeling_roberta_sentiment.RobertaForSentimentClassification"
}
}