Publishing model weights bin

config.json

@@ -3,18 +3,32 @@
     "Seq2SeqTransformer"
   ],
   "model_type": "encoder-decoder",
-  "num_encoder_layers": 3,
-  "num_decoder_layers": 3,
-  "emb_size": 256,
-  "nhead": 8,
-  "src_vocab_size": 318,
-  "tgt_vocab_size": 318,
-  "dim_feedforward": 512,
-  "dropout": 0.1,
-  "special_tokens": {
-    "pad_token": "<pad>",
-    "bos_token": "<bos>",
-    "eos_token": "<eos>",
-    "unk_token": "<unk>"
+  "encoder": {
+    "num_layers": 3,
+    "hidden_size": 256,
+    "num_attention_heads": 8,
+    "intermediate_size": 512,
+    "dropout": 0.1,
+    "vocab_size": 318,
+    "special_tokens": {
+      "pad_token": "<pad>",
+      "bos_token": "<bos>",
+      "eos_token": "<eos>",
+      "unk_token": "<unk>"
+    }
+  },
+  "decoder": {
+    "num_layers": 3,
+    "hidden_size": 256,
+    "num_attention_heads": 8,
+    "intermediate_size": 512,
+    "dropout": 0.1,
+    "vocab_size": 318,
+    "special_tokens": {
+      "pad_token": "<pad>",
+      "bos_token": "<bos>",
+      "eos_token": "<eos>",
+      "unk_token": "<unk>"
+    }
   }
 }