Update config.json

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