diff --git a/paddle2.0_docs/pretrained_word_embeddings/pretrained_word_embeddings.ipynb b/paddle2.0_docs/pretrained_word_embeddings/pretrained_word_embeddings.ipynb
index b1ae5b8e..fc416889 100644
--- a/paddle2.0_docs/pretrained_word_embeddings/pretrained_word_embeddings.ipynb
+++ b/paddle2.0_docs/pretrained_word_embeddings/pretrained_word_embeddings.ipynb
@@ -171,10 +171,10 @@
     "def vectorizer(input, label=None, length=2000):\r\n",
     "    if label is not None:\r\n",
     "        for x, y in zip(input, label):\r\n",
-    "            yield np.array((x + [0]*length)[:2000]).astype('int64'), np.array([y]).astype('int64')\r\n",
+    "            yield np.array((x + [0]*length)[:length]).astype('int64'), np.array([y]).astype('int64')\r\n",
     "    else:\r\n",
     "        for x in input:\r\n",
-    "            yield np.array((x + [0]*length)[:2000]).astype('int64')"
+    "            yield np.array((x + [0]*length)[:length]).astype('int64')"
    ]
   },
   {