dilip025 commited on
Commit
6a18d5d
·
verified ·
1 Parent(s): ebad92f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -6
config.json CHANGED
@@ -1,17 +1,16 @@
1
  {
2
- "architectures": [
3
- "RobertaForSentimentClassification"
4
- ],
5
  "vocab_size": 30000,
6
  "hidden_size": 512,
7
  "num_attention_heads": 8,
8
  "num_hidden_layers": 6,
9
  "intermediate_size": 2048,
10
- "hidden_dropout_prob": 0.1,
11
  "max_position_embeddings": 128,
12
  "num_labels": 5,
13
- "model_type": "roberta",
14
  "auto_map": {
 
15
  "AutoModelForSequenceClassification": "modeling_roberta_sentiment.RobertaForSentimentClassification"
16
  }
17
- }
 
1
  {
2
+ "architectures": ["RobertaForSentimentClassification"],
3
+ "model_type": "roberta-sentiment",
 
4
  "vocab_size": 30000,
5
  "hidden_size": 512,
6
  "num_attention_heads": 8,
7
  "num_hidden_layers": 6,
8
  "intermediate_size": 2048,
 
9
  "max_position_embeddings": 128,
10
  "num_labels": 5,
11
+ "hidden_dropout_prob": 0.1,
12
  "auto_map": {
13
+ "AutoConfig": "modeling_roberta_sentiment.RobertaSentimentConfig",
14
  "AutoModelForSequenceClassification": "modeling_roberta_sentiment.RobertaForSentimentClassification"
15
  }
16
+ }