EmmanuelM1 commited on
Commit
4f2c82a
·
verified ·
1 Parent(s): 5764531

Update config.json

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