EmmanuelM1 commited on
Commit
b1d9f98
1 Parent(s): 4f2c82a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "architectures": ["RobertaForSentencePrediction"],
3
  "model_type": "roberta",
4
  "num_classes": 2,
5
  "num_attention_heads": 16,
 
1
  {
2
+ "architectures": ["RobertaForSequenceClassification"],
3
  "model_type": "roberta",
4
  "num_classes": 2,
5
  "num_attention_heads": 16,