Update config.json

2020-01-31 23:00:24 +00:00 · 2020-01-31 23:00:24 +00:00 · d9fd1c824e
parent 2def212e96
commit d9fd1c824e
1 changed files with 3 additions and 0 deletions
--- a/config.json
+++ b/config.json
@ -1,4 +1,7 @@
 {
+  "architectures": [
+    "DistilBertForSequenceClassification"
+  ],
  "activation": "gelu",
  "attention_dropout": 0.1,
  "dim": 768,