Merge pull request yl4579#42 from yl4579/yl4579-patch-1

Update train.ipynb
Conv-AI · Jan 8, 2024 · 4198635 · 4198635
2 parents 592293a + 85bd638
commit 4198635
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.ipynb b/train.ipynb
@@ -117,7 +117,7 @@
     "    \n",
     "    bert = AlbertModel(albert_base_configuration)\n",
     "    bert = MultiTaskModel(bert, \n",
-    "                          num_vocab=max([m['token'] for m in token_maps.values()]), \n",
+    "                          num_vocab=1 + max([m['token'] for m in token_maps.values()]), \n",
     "                          num_tokens=config['model_params']['vocab_size'],\n",
     "                          hidden_size=config['model_params']['hidden_size'])\n",
     "    \n",